小 A 面前有 n 堆石子排成一排,每堆石子的数量从左到右依次为 a1,a2,…,an。
小 B 面前有 m 堆石子排成一排,每堆石子的数量从左到右依次为 b1,b2,…,bm。
两人面前的石子总数相同,即 a1+a2+…+an=b1+b2+…+bm。
每个人都可以对自己面前的石子堆进行任意次合并操作,两个人的操作次数可以不同。
合并操作指将连续的若干堆相邻石子合并为一堆。
请你确定一个最大的整数 kk,满足:
- 小 AA 停止所有操作后,面前恰好有 kk 堆石子,每堆石子的数量从左到右依次为 a′1,a′2,…,a′k。
- 小 BB 停止所有操作后,面前恰好有 kk 堆石子,每堆石子的数量从左到右依次为 b′1,b′2,…,b′k。
- 对于 1≤i≤k,a′i=b′i 均成立。
输出这个 k 的最大可能值。
输入格式
第一行包含两个整数 n,m。
第二行包含 n 个整数 a1,a2,…,an。
第三行包含 m 个整数 b1,b2,…,bm。
输出格式
一个整数,表示 k 的最大可能值。
数据范围
前三个测试点满足 1≤n,m≤10。
所有测试点满足 1≤n,m≤1e5,1≤ai,bi≤1e6,a1+a2+…+an=b1+b2+…+bm≤1e6
输入样例1:
7 6
2 5 3 1 11 4 4
7 8 2 4 1 8
输出样例1:
3
输入样例2:
3 3
1 10 100
1 100 10
输出样例2:
2
输入样例3:
1 4
4
1 1 1 1
输出样例3:
1
分析:这道题用到了贪心的思想,也就是说如果此堆石子中a=b就立刻换成下一堆,这样就能保证石子对数最多。
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N],b[N];
int n,m,ans;
int suma,sumb;
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
for(int j=1;j<=m;j++) cin>>b[j];
int i=0,j=0;
while(i<=n&&j<=m)
{
if(suma<sumb)
{
i++;
suma+=a[i];
}
else if(suma>sumb)
{
j++;
sumb+=b[j];
}
else
{
i++,j++;
ans++;
suma=a[i],sumb=b[j];
}
}
cout<<ans-1;
}