1. 程式人生 > >D. Vasya and Arrays

D. Vasya and Arrays

比較大小 最大數 題意 cout bit with 多少 無法 else if

鏈接

[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;
}
從頭和尾兩個位置判斷是否相等,相等繼續如果不相等統計兩個數組區間的長度,在比較大小,進行到最後
具體看代碼

代碼

D. Vasya and Arrays