题目概述:一次操作选定一个区间,将区间内数的总和赋给区间内的某个数并清零区间内的其他数。判断\(a\)序列是否能变成\(b\)序列,如果能则输出最小操作数。
满分做法:
用双指针记录现在到了&a&序列的位置和\(b\)序列的位置,每次\(b[i]\)不等于0时,\(ans++\),如果本身a[i]本身就符合条件再减回来。剩下的加就行了,如果加不够就输出-1.
另外需要开始判总和是否相等。
#include<queue>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxm=1e+7;
int t,n,ans;
int a[maxm],b[maxm];
int solve()
{
int r=1;
ll sum=0;
ans=0;
for(int i=1;i<=n;i++)
{
if(!b[i]) continue;
ans++;
int l=r;
sum=0;
while((sum<b[i])&&(r<=n)) sum+=a[r++];
if(sum!=b[i]){
return -1;
}
if(i>=l&&i<=r-1&&a[i]==b[i]) ans--;//;如果段只包含了自己并且位置相同,我们就不需要浪费操作在它身上
}
return ans;
}
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
ll sum1=0,sum2=0;
for(int i=1;i<=n;i++)
scanf("%d",a+i),sum1+=a[i];
for(int i=1;i<=n;i++)
scanf("%d",b+i),sum2+=b[i];
if(sum1!=sum2)//如果b[i]后面有连续的0,下面的就不能输出-1了
{
printf("-1\n");
continue;
}
printf("%d\n",solve());
}
return 0;
}