Comet OJ模拟赛 Day1

A.修行

题目概述:一次操作选定一个区间,将区间内数的总和赋给区间内的某个数并清零区间内的其他数。判断\(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;  
}

猜你喜欢

转载自www.cnblogs.com/lihan123/p/11828940.html
今日推荐