結果

問題 No.1770 N言っちゃダメゲーム (6)
ユーザー pes_magicpes_magic
提出日時 2021-12-02 22:27:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 50 ms / 2,000 ms
コード長 1,182 bytes
コンパイル時間 876 ms
コンパイル使用メモリ 84,972 KB
実行使用メモリ 25,928 KB
最終ジャッジ日時 2024-07-05 02:10:18
合計ジャッジ時間 2,723 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>

using namespace std;

vector<int> solve(int N, int K){
    vector<vector<int>> pushVal(N);
    vector<vector<int>> popVal(N);
    vector<int> lose(N, 0);
    set<int> loseIdx;
    for(int i=N-1;i>0;i--){
        for(auto& t : pushVal[i]) loseIdx.insert(t);
        for(auto& t : popVal[i]) loseIdx.erase(t);
        if(loseIdx.empty()){
            lose[i] = -1;
            pushVal[i-1].push_back(i);
            if(i-K-1 > 0) popVal[i-K-1].push_back(i);
        } else if(loseIdx.size() == 1){
            int loseK = *loseIdx.begin() - i;
            lose[i] = loseK;
            if(i-loseK > 0) pushVal[i-loseK].push_back(i);
            if(i-loseK-1 > 0) popVal[i-loseK-1].push_back(i);
        } else {
            lose[i] = 0;
        }
    }
    vector<int> res;
    for(int i=1;i<=K;i++){
        if(lose[i] == -1 || lose[i] == i) res.push_back(i);
    }
    return res;
}

int main(){
    int N, K;
    while(cin >> N >> K){
        auto res = solve(N, K);
        if(res.empty()){
            cout << 0 << endl;
        } else {
            for(auto& t : res) cout << t << endl;
        }
    }
}
0