第十四届蓝桥杯大赛软件赛决赛 C/C++ 大学 B 组 试题 D: 合并数列

[蓝桥杯 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$ 行，一个整数。

【样例输入】

4 3
1 2 3 4
1 5 4

【样例输出】

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$。

#include<bits/stdc++.h>
using namespace std;
int main()
{
    
    
	list<int> a,b;
	int n,m;cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
    
    
		int x;cin>>x;
		a.push_back(x);
	}
	for(int i=1;i<=m;i++)
	{
    
    
		int x;cin>>x;
		b.push_back(x);
	}
	int cnt=0;
	while(!a.empty() && !b.empty())
	{
    
    
		if(a.front()==b.front()) 
		{
    
    
			a.pop_front();
			b.pop_front();
		}
		else if(a.front()<b.front())
		{
    
    
			int a1=a.front();a.pop_front();
			int a2=a.front();a.pop_front();
			a.push_front(a1+a2);
			cnt++;
		}
		else if(a.front()>b.front())
		{
    
    
			int b1=b.front();b.pop_front();
			int b2=b.front();b.pop_front();
			b.push_front(b1+b2);
			cnt++;
		}
	}
	cout<<cnt;
	
    return 0;
}