結果

問題 No.349 干支の置き物
ユーザー gam0022
提出日時 2016-05-01 21:34:25
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 2,115 bytes
コンパイル時間 1,284 ms
コンパイル使用メモリ 84,852 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-07 14:35:35
合計ジャッジ時間 1,829 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <list>
#include <deque>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;

void print_map(map<string, int> m) {
    for(auto itr = m.begin(); itr != m.end(); ++itr) {
        std::cout << "key = " << itr->first
            << ", val = " << itr->second << "\n";
    }
}

void print_vector(vector<int> v) {
    for(auto itr = v.begin(); itr != v.end(); ++itr) {
        std::cout << *itr << ", ";
    }
    cout << endl;
}

vector<int> get_vals_from_map(map<string, int> m) {
    vector<int> vals;

    for(auto itr = m.begin(); itr != m.end(); ++itr) {
        vals.push_back(itr->second);
    }

    return vals;
}

void check(vector<int> counts) {
    int max = 0;
    int rest = 0;

    int n = counts.size();
    for(int i = 0; i < n; ++i) {
        int count = counts[i];
        if (i < n - 1) {
            rest += count;
        } else {
            max += count;
        }
    }

    //cout << "max: " << max << endl;
    //cout << "rest:" << rest << endl;

    if ( max == rest || max == rest - 1 || max == rest + 1) {
        cout << "YES" << endl;
        return;
    } else if (max + 1 > rest) {
        cout << "NO" << endl;
    } else {
        counts.pop_back();
        int n = counts.size();
        //print_vector(counts);
        int i = n - 1;
        while(max > 0) {
            i = i % n;
            if (counts[i] > 0) {
                --counts[i];
                --max;
            }
            --i;
        }
        //print_vector(counts);
        counts.erase(remove(counts.begin(), counts.end(), 0), counts.end());
        sort(counts.begin(), counts.end());
        //print_vector(counts);
        //cout << "--" << endl;
        check(counts);
    }
}

int main(){
    int n;
    map<string, int> etos;
    cin >> n;
    for (int i = 0; i < n; i++) {
        string eto;
        cin >> eto;
        ++etos[eto];
    }

    //print_map(etos);
    vector<int> counts = get_vals_from_map(etos);
    sort(counts.begin(), counts.end());

    check(counts);

    return 0;
}
0