Atcode B - Colorful Hats（思维）

题目链接：http://agc016.contest.atcoder.jp/tasks/agc016_b

题解：挺有意思的题目主要还是模拟出最多有几种不可能的情况，要知道ai的差距不能超过1这个想想就知道，然后再好好想一下，想想可能会有点麻烦

聪明的人应该一下就想出来了，反正我想了挺久。

#include <iostream>

#include <cstring>

using namespace std;

const int M = 1e5 + 10;

int a[M];

int main() {

    int n;

    cin >> n;

    for(int i = 1 ; i <= n ; i++) cin >> a[i];

    int l = a[1] , r = a[1];

    for(int i = 2 ; i <= n ; i++) {

        l = min(a[i] , l);

        r = max(a[i] , r);

    }

    if(r - l > 1) cout << "No" << endl;

    else {

        if(l == r) {

            int gg = n - l;

            if(gg == 1) cout << "Yes" << endl;

            else {

                if(n == 1 || gg >= l) cout << "Yes" << endl;

                else cout << "No" << endl;

            }

        }

        else {

            int cnt1 = 0 , cnt2 = 0;

            for(int i = 1 ; i <= n ; i++) {

                if(a[i] == l) cnt1++;

                if(a[i] == r) cnt2++;

            }

            if(cnt1 >= r) cout << "No" << endl;

            else {

                int gg = n - r;

                if(gg >= r - cnt1) cout << "Yes" << endl;

                else cout << "No" << endl;

            }

        }

    }

    return 0;

}

巴特西

Atcode B - Colorful Hats（思维）

最新文章

热门文章