[agc016b]Colorful Hats

前言

简单的讨论题。

题意

n个人每个人戴着一种颜色的帽子。
第i个人说看到了ai种颜色的帽子（自己不能看到自己戴的帽子）。
问是否全部都在说真话？

做法

假设一共有k种颜色的帽子。
一个独一无二的颜色会说k-1其余说k。
如果最大的ai减最小的ai大于1一定有假话。
然后考虑最大值不等于最小值的情况。
对于说k-1的当然都独一无二，剩余的颜色每种至少分配给两个人。
如果最大值等于最小值，有两种情况，全都独一无二，或者全都不独一无二，前者一定是n-1，后者类似判断即可。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
int i,j,k,l,t,n,m,mx,mi,cnt;
bool czy;
int main(){
    scanf("%d",&n);
    czy=1;
    mi=n+1;mx=0;
    fo(i,1,n){
        scanf("%d",&t);
        //mi=min(mi,t);
        if (t<mi) mi=t,cnt=1;
        else if (t==mi) cnt++;
        if (i>1&&czy&&t!=mx) czy=0;
        mx=max(mx,t);
    }
    if (czy){
        if (mx==n-1) printf("Yes\n");
        else{
            k=mx;
            if (k*2<=n) printf("Yes\n");
            else printf("No\n");
        }
    }
    else{
        if (mx-mi>1) printf("No\n");
        else{
            k=mx;
            if (cnt<k&&n-cnt>=(k-cnt)*2) printf("Yes\n");
            else printf("No\n");
        }
    }
}