智乃想考一道鸽巢原理 鸽巢原理 （抽屉原理） 杭电oj 吃糖果

void solve()
{
    cin >> n;
    S = N = 0;
    for(int i = 1; i <= n; ++i)
    {
        cin >> x;
        N = max(N, x);
        S += x;
    }
    S -= N;
    cout << (S >= N - 1 ? "Yes" : "No") << endl;
}

附上 买糖果代码：

#include <stdio.h>
#define M 1000000
#define max(a,b) a>b?a:b
int i,n,T,MAX;
int a[M];
int main()
{
    scanf("%d",&T);
    while(T--)
    {

        long long sum=0;
        scanf("%d",&n);
        MAX=0;
        for(i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];        
            MAX=max(MAX,a[i]); 
        }
        sum=sum-MAX+1; 
        if(sum>=MAX) 
        {
            printf("Yes\n");
        }
        else 
        {
            printf("No\n");
        }
    }
    return 0;
}

再来看原来题目的：

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 5;
const int INF = 0x7fffffff;
const double eps = 1e-5;

//t表示测试用例的数量
// n表示每种颜色的小球数量
// a是一个数组用于存储每种颜色的小球数量
// max1和max2分别表示最大值和次大值
// maxa表示当前考虑的最大值
// mi表示最大值的索引
// sum表示所有小球数量的总和。
int t,n,a[N];
int max1,max2,maxa,mi;
long long sum=0;

//首先，通过输入获取每种颜色的小球数量
// 并计算总和。然后，找到最大值和次大值
// 接下来，根据鸽巢原理进行判断
// 如果某种颜色的小球数量减去最大值和次大值之和大于等于次大值减一
// 则输出1，否则输出0。最后，输出结果。
void solve() {
    max1=max2=-INF;
    sum=0;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
        sum+=a[i];
        if(a[i]>max1){
            max1=a[i];//求最大值
            mi=i;
        }
    }
    if(n==1) {//特判1
        cout<<1<<endl;
        return;
    }
    for(int i=0;i<n;i++){
        if(i!=mi) max2=max(max2,a[i]);//求第二大值
    }
    for(int i=0;i<n;i++){//重点 鸽巢原理
        maxa=(max1==a[i])?max2:max1;
        if(sum-a[i]-maxa>=maxa-1) {
            if(a[i]>((sum-a[i])%2)) cout<<1<<' ';
            else cout<<0<<' ';
        }
        else {
            if(a[i]>(2*maxa-sum+a[i])) cout<<1<<' ';
            else cout<<0<<' ';
        }
    }
    cout<<endl;
}

int main() {
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin>>t;
    while(t--){
        solve();
    }
    return 0;
}