题目
HOHO,终于从Speakless手上赢走了所有的糖果,是Gardon吃糖果时有个特殊的癖好,就是不喜欢将一样的糖果放在一起吃,喜欢先吃一种,下一次吃另一种,这样;可是Gardon不知道是否存在一种吃糖果的顺序使得他能把所有糖果都吃完?请你写个程序帮忙计算一下。
Input
第一行有一个整数T,接下来T组数据,每组数据占2行,第一行是一个整数N(0<N<=1000000),第二行是N个数,表示N种糖果的数目Mi(0<Mi<=1000000)。
Output
对于每组数据,输出一行,包含一个"Yes"或者"No"。
Sample
Input
2
3
4 1 1
5
5 4 3 2 1
Output
No
Yes
思路
本题为典型的鸽巢原理问题,n种糖果个数的最大值max视为鸽子,其余糖果为隔板,至少需要max-1个隔板,糖果总个数为s;如果max-1>s-max,那么一定不可能以这样的顺序把所有糖果均吃完;反之,可以吃完。注意数据范围需要开long long,在每组数据之前都要初始化数组。
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+9;
ll a[N];
int main(){
int t;
cin>>t;
while(t--){
memset(a,0,N);
int n,i;
ll s=0,max;
cin>>n;
for(i=0;i<n;++i){
cin>>a[i];
s+=a[i];
}
bool f=true;
max=*max_element(a,a+n);
if(max-1>s-max) f=false;
if(f) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
return 0;
}