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

posted @ 2019-04-14 21:57  努力进步的肥宅yxc  阅读(102)  评论(0编辑  收藏  举报