Date:2022.01.12
题意:给定A、B两个序列,连续子序列的元素可合并,合并若干次后将两个序列变成相同的,让长度最大;不能合并输出-1。
思路:一眼双指针贪心,因为只能合并连续的子序列,如果一个序列能合并,该序列任意位置的一个元素都能在某处合并或者保持它本身不变。因此我们规定让前面的先合并,这样贪心一定保证正确:如果碰到不能合并的则指针卡住,最后判断两个指针是否都>序列长度,>则表示匹配,任一指针<=序列长度则表示无法匹配输出-1。
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int N = 3e5+10;
typedef long long LL;
LL n,m,t;
LL a[N],b[N];
int main()
{
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>n;for(int i=1;i<=n;i++) cin>>a[i];
cin>>m;for(int i=1;i<=m;i++) cin>>b[i];
LL l=1,l2=1,res1=a[1],res2=b[1],ans=0;
while(l<=n&&l2<=m)
{
//cout<<res1<<' '<<res2<<endl;
if(res1==res2) {l++,l2++;res1=a[l],res2=b[l2];ans++;}
else if(res1>res2) {l2++;res2+=b[l2];}
else if(res1<res2) {l++;res1+=a[l];}
}
if(l>n&&l2>m) cout<<ans<<endl;
else cout<<"-1"<<endl;
return 0;
}