AtCoder Grand Contest 016（B - Colorful Hats）

B - Colorful Hats

 

题意：有n个颜色，第i个数字表示第i个颜色不同的数量，求输入的数据是否有矛盾。

一个数学题目，可惜推不出来。只能看大佬的了。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int main(){
    int n;
    cin >> n;
    vector<int> a(n);
    int ans = 0;
    for(int i = 0; i < n; i ++) scanf("%d",&a[i]);
    sort(a.begin(),a.end());
    int minn = a[0];
    int maxn = a.back();
    if(minn == maxn){
        if(minn*2 <= n || minn == n-1) ans = 1;
    }
    if(maxn == minn +1 ){
        int cnt = 0;
        for(int i = 0; a[i] == minn; i++) cnt++;
        if(maxn > cnt && cnt + 2*(maxn-cnt) <= n) ans = 1;
    }
    printf("%s\n",ans?"Yes":"No");
    return 0;
}