【题解-洛谷】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 n = m n=m 且对于任意下标 i i i 满足 a i = b i a_i = b_i ai=bi。请计算至少需要多少次合并操作可以完成小明的目标。

输入格式

输入共 3 3 3 行。

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

第二行为 n n n 个由空格隔开的整数 a 1 , a 2 , ⋯ , a n a_1, a_2, \cdots, a_n a1,a2,⋯,an。

第三行为 m m m 个由空格隔开的整数 b 1 , b 2 , ⋯ , b m b_1, b_2, \cdots, b_m b1,b2,⋯,bm。

输出格式

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

输入输出样例 #1

输入 #1

复制代码

4 3
1 2 3 4
1 5 4

输出 #1

复制代码

说明/提示

样例说明

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

评测用例规模与约定

对于 20 % 20\% 20% 的数据，保证 n , m ≤ 10 3 n,m \le 10^3 n,m≤103。
对于 100 % 100\% 100% 的数据，保证 n , m ≤ 10 5 n, m \le 10^5 n,m≤105， 0 < a i , b i ≤ 10 5 0 < a_i, b_i \le 10^5 0<ai,bi≤105。

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

代码

cpp 复制代码

#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;
}