AGC016B Colorful Hats
题意简述:有n个人每个人戴着一种颜色的帽子,第i个人说看到\(a_i\)种颜色的帽子(自己不能看到自己戴的帽子)。问是否全部都在说真话?
可以,我又不会做。orz wzd
首先有一个非常显然的东西,这个我自己想出来了:就是合法情况下,\(max_{a_i}-min_{a_i}<=1\)因为每个人要么自己带着独有的帽子,要么带着非独有的帽子。
然后我们再分情况考虑\(max_{a_i}-min_{a_i}=1\)与\(max_{a_i}=min_{a_i}\)。
如果\(max_{a_i}-min_{a_i}=1\),那么除了最小值要独有,其他最大值都至少要两顶颜色相同的帽子,搞一搞就ok了
如果\(max_{a_i}=min_{a_i}\),那么要么所有人都有独有颜色的帽子,要么都至少有两顶颜色相同帽子。
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int cnt,mx,mn=1e9,n,t[N];
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&t[i]),mx=max(mx,t[i]),mn=min(mn,t[i]);
if(mx-mn>1){puts("No");return 0;}
if(mx==mn){
if(mx==n-1){puts("Yes");return 0;}
if(mx*2<=n)puts("Yes");
else puts("No");
return 0;
}
for(int i=1;i<=n;i++)
if(t[i]==mn)++cnt;
if(cnt<mx&&cnt+(mx-cnt)*2<=n)puts("Yes");
else puts("No");
}