题目描述
小杨有一个包含n个正整数的序列a,他认为一个序列是平衡的当且仅当存在一个正整数 i(1<=i<n )使得序列第1个到第i个数字的总和等于第i+1个到第n个数字的总和。 小杨想请你判断a序列 是否是平衡的
输入
第一行包含一个正整数t,代表测试用例组数。 接下来是t组测试用例。对于每组测试用例,一共两行。 第一行包含一个正整数n,代表序列长度。 第二行包含n个正整数,代表序列a
输出
对于每组测试用例,如果序列a是否是平衡的,输出 Yes,否则输出No.
样例输入
3
3
1 2 3
4
2 3 1 4
5
1 2 3 4 5
样例输出
Yes
Yes
No
提示
对于第一组测试用例,令i=2 ,则有1+2=3 ,因此序列是平衡的;
对于第二组测试用例,令i=2 ,则有2+3=1+4 ,因此序列是平衡的;
对于第三组测试用例,不存在满足要求的 。
对于全部数据,保证有1<=t<=100,1<=n,ai<=10000 。
代码
#include <bits/stdc++.h>
using namespace std;
int main(){
int t,n,a[10005];
cin>>t;
for(int i=1;i<=t;i++){
cin>>n;
for(int j=1;j<=n;j++){
cin>>a[j];
a[j]+=a[j-1];
}
bool b=false;
for(int j=1;j<=n-1;j++){
if(a[n]-a[j]==a[j]){
b=true;
}
}
if(b)cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
return 0;
}
如果时间超限,可以在main函数开头加入下面代码加速,带只能用cout,cin,另外,请使用'\n'而不是 endl ,因为endl默认会增加刷新操作,而导致输出缓冲失效,降低效率。
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);