[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");        }    }}