合并数列（蓝桥杯）

WA哭我了~

主要还是前缀和+双指针，算法不难，但是很注重细节，被细节卡了两个小时，主要讲讲代码实现思路和为什么会WA。（借鉴了大佬题解，哭了，基础题WA成狗了）

0合并数列 - 蓝桥云课

问题描述

小明发现有很多方案可以把一个很大的正整数拆成若干正整数的和。他采取了其中两种方案，分别将他们列为两个数组 {a1,a2,...,an}{a1​,a2​,...,an​} 和 {b1,b2,...,bm}{b1​,b2​,...,bm​}。两个数组的和相同。

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

输入格式

输入共 33 行。

第一行为两个正整数 nn, mm。

第二行为 nn 个由空格隔开的整数 a1,a2,...,ana1​,a2​,...,an​。

第三行为 mm 个由空格隔开的整数 b1,b2,...,bmb1​,b2​,...,bm​。

输出格式

输出共 11 行，一个整数。

样例输入

4 3
1 2 3 4
1 5 4

样例输出

1

样例说明

只需要将 a2a2​ 和 a3a3​ 合并，数组 aa 变为 {1,5,4}{1,5,4}，即和 bb 相同。

评测用例规模与约定

对于 20%20% 的数据，保证 nn, m≤103m≤103。

对于 100%100% 的数据，保证 nn, m≤105m≤105，0

运行限制

语言	最大运行时间	最大运行内存
C++	1s	256M
C	1s	256M
Java	2s	256M
Python3	3s	256M
PyPy3	3s	256M
Go	3s	256M
JavaScript	3s	256M

 先讲讲WA的答案吧~

#include
using namespace std;
#define int long long
#define endl '\n'
//考点：前缀和 + 双指针

int n,m;
int a[100010],b[100010];
signed main(){
  cin>>n>>m;
  for(int i=1;i<=n;i++) cin>>a[i];
  for(int i=1;i<=m;i++) cin>>b[i];
  int ans=0;
  int suma=0,sumb=0;
  int i=1,j=1;
  while(i<=n&&j<=m){
    if(suma==sumb){
      suma=a[i];i++;
      sumb=b[j];j++;
    }
    if(suma>sumb){
      sumb+=b[j];j++;ans++;
    }
    if(suma

 

主要实现思路：

因为两个数组a和b的顺序是固定的，我们只需要按照顺序进行合并即可，我们知道两个数组的和是相等的，同理可得，如果数组进行合并后的和相等，那么说明两个数组剩下的和也是相等的所以每次都是对数组进行区间求和，让两个数组相等即可。

所以定义了i=1，j=1。从最开始时进行判断，如果suma==sumb，那么放弃前面的相同和，suma和sumb都读取下一个数，读取结束后进行判断，如果suma小，suma再合并后一个数，i++。否则如果sumb小，sumb再合并下一个数，j++。注意这里我们不需要考虑suma==sumb的情况，这个在下一个循环考虑。

其实上述实现思路和正解思路完全相同，区别在于，如果进行else if(suma

但是其实仔细观察可以发现，这里其实会多进行一次操作，即对n+1和m+1进行操作，由于两个数组前缀和是一样的，所以多进行一次操作也不会有太大影响，所以将WA代码的循环边界扩大一个单位也能得到一样的效果。

正解如下：

#include
using namespace std;
#define int long long
#define endl '\n'
//考点：前缀和 + 双指针

int n,m;
int a[100010],b[100010];
signed main(){
  cin>>n>>m;
  for(int i=1;i<=n;i++) cin>>a[i];
  for(int i=1;i<=m;i++) cin>>b[i];
  int ans=0;
  int suma=0,sumb=0;
  int i=1,j=1;
  while(i<=n+1&&j<=m+1){
    if(suma==sumb){
      suma=a[i];i++;
      sumb=b[j];j++;
    }
    if(suma>sumb){
      sumb+=b[j];j++;ans++;
    }
    if(suma