题目:
输入例子1:
1
6
1 2 3 4 5 6
输出例子1:
NO
输入例子2:
1
4
4 4 5 3
输出例子2:
YES
说明:
若答案序列是由前面部分数字加上后面部分数字构成,那么等价于中间存在一段连续序列。若答案序列是前面的某一段,那么等价于后面的某一段。即s1=s2,也就是说,我们只需要考虑前面或者中间是否存在一段连续序列使得和恰好为总序列和的二分之一即可。
代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
signed main(){
int t;
cin>>t;
while(t--){
int n;
int a[maxn]={0};
int sum=0;
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<=n;i++){
sum+=a[i];
}
int pos=1,s=0;
bool flag=false;
if(sum%2==0){ //为偶数才可能被平分成两段
for(int i=1;i<=n;i++){
s+=a[i];
while(s>sum/2) s-=a[pos++];
if(s==sum/2){
flag=true;
break;
}
}
}
if(flag){
cout<<"YES"<<endl;
}else{
cout<<"NO"<<endl;
}
}
return 0;
}