D. Vasya and Arrays

Posted 2021-01-16 mch5201314

链接

[http://codeforces.com/contest/1036/problem/D]

题意

给你两个数组长度分别为n,m;
有这么一种操作，用某个数组的某个子区间元素之和代替这个子区间，这样使得数组长度减少，两个数组都可以进行
问你最后两个数组一摸一样的时候，最大数组长度是多少？如果无法使两个数组一摸一样输出-1

分析

先判断开始数组总和是否相等，不相等是不可能最后两个数组一摸一样的#include<bits/stdc++.h>
using namespace std;

define ll long long

const ll N=3e5+10;
ll a[N],b[N];
ll n,m;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
//freopen("in.txt","r",stdin);
cin>>n;
ll sum1=0,sum2=0;
for(int i=0;i<n;i++)
{
cin>>a[i];
sum1+=a[i];
}
cin>>m;
for(int i=0;i<m;i++)
{
cin>>b[i];
sum2+=b[i];
}
if(sum1!=sum2){
cout<<-1<<endl;
return 0;
}
int cnt=0;
if(a[0]==b[0])
{
int i=1,j=1;
sum1=a[1],sum2=b[1];
while(i<n&&j<m){
if(sum1<sum2){
sum1+=a[i+1];
cnt++;
i++;
}
else if(sum1>sum2){
sum2+=b[j+1];
j++;
}
else {
sum1=a[i+1],sum2=b[j+1];
i++,j++;
}
}
}
else if(a[n-1]==b[m-1]){
int i=n-1,j=m-1;
sum1=a[n-1],sum2=b[m-1];
while(i>-1&&j>-1){
if(sum1<sum2){
sum1+=a[i-1];
cnt++;
i--;
}
else if(sum2<sum1){
sum2+=b[j-1];
j--;
}
else {
sum1=a[i-1],sum2=b[j-1];
i--,j--;
}
}
}
else{
int i=0,j=0;
sum1=a[0],sum2=b[0];
while(i<n&&j<m){
if(sum1<sum2){
sum1+=a[i+1];
cnt++;
i++;
}
else if(sum1>sum2)
{
sum2+=b[j+1];
j++;
}
else {
sum1=a[i+1],sum2=b[j+1];
i++,j++;
}
}
}
//cout<<cnt<<endl;
cout<<n-cnt<<endl;
return 0;
}
从头和尾两个位置判断是否相等，相等继续如果不相等统计两个数组区间的长度，在比较大小，进行到最后
具体看代码

代码