【题解-洛谷】P9422 [蓝桥杯 2023 国 B] 合并数列

题目：P9422 [蓝桥杯 2023 国 B] 合并数列

题目描述

小明发现有很多方案可以把一个很大的正整数拆成若干正整数的和。他采取了其中两种方案，分别将他们列为两个数组 { a 1 , a 2 , ⋯ a n } \{a_1, a_2, \cdots a_n\} {a1​,a2​,⋯an​} 和 { b 1 , b 2 , ⋯ b m } \{b_1, b_2, \cdots b_m\} {b1​,b2​,⋯bm​}。两个数组的和相同。

定义一次合并操作可以将某数组内相邻的两个数合并为一个新数，新数的值是原来两个数的和。小明想通过若干次合并操作将两个数组变成一模一样，即 $n=m$ 且对于任意下标 $i$ 满足 $a_i=b_i$。请计算至少需要多少次合并操作可以完成小明的目标。

输入格式

输入共 $3$ 行。

第一行为两个正整数 $n,m$。

第二行为 $n$ 个由空格隔开的整数 $a_1,a_2,\cdots ,a_n$。

第三行为 $m$ 个由空格隔开的整数 $b_1,b_2,\cdots ,b_m$。

输出格式

输出共 $1$ 行，一个整数。

输入输出样例 #1

输入 #1

4 3
1 2 3 4
1 5 4

输出 #1

1

说明/提示

样例说明

只需要将 $a_2$ 和 $a_3$ 合并，数组 $a$ 变为 { 1 , 5 , 4 } \{1,5,4\} {1,5,4}，即和 $b$ 相同。

评测用例规模与约定

• 对于 $20\%$ 的数据，保证 $n,m\le 10^3$。
• 对于 $100\%$ 的数据，保证 $n,m\le 10^5$，$0<a_i,b_i\le 10^5$。

第十四届蓝桥杯大赛软件赛决赛 C/C++ 大学 B 组 D 题

代码

#include<iostream>

using namespace std;

typedef long long LL;

const int Maxn = 1e5 + 10;

LL n, m, a[Maxn], b[Maxn], q1[Maxn], q2[Maxn], hh1, tt1 = -1, hh2, tt2 = -1, sum;

void insert(int id, LL x){
    if(id == 1){
        q1[++ tt1] = x;
    }
    else{
        q2[++ tt2] = x;
    }
}

void dele(int id){
    if(id == 1){
        hh1 ++;
    }
    else{
        hh2 ++;
    }
}

bool isempty(int id){
    if(id == 1){
        return hh1 > tt1;
    }
    else{
        return hh2 > tt2;
    }
}

int main(){
    cin >> n >> m;
    for(int i = 0; i < n; i ++){
        int x;
        cin >> x;
        insert(1, x);
    }
    for(int i = 0; i < m; i ++){
        int x;
        cin >> x;
        insert(2, x);
    }

    while(!isempty(1) || !isempty(2)){
        LL a = q1[hh1];
        dele(1);
        LL b = q2[hh2];
        dele(2);

        while(a < b){
            a += q1[hh1];
            dele(1);
            sum ++;
        }
        while(b > a){
            b += q2[hh2];
            dele(2);
            sum ++;
        }
    }

    cout << sum;
    return 0;
}

结果