結果

問題 No.1770 N言っちゃダメゲーム (6)
ユーザー trineutrontrineutron
提出日時 2020-06-02 18:56:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 22 ms / 2,000 ms
コード長 1,365 bytes
コンパイル時間 1,828 ms
コンパイル使用メモリ 81,120 KB
最終ジャッジ日時 2025-01-10 20:36:32
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cassert>

int main() {
    int n, k;
    std::cin >> n >> k;
    assert(1 <= k and k < n and n <= 200000);
    if (k == 1) {
        std::cout << 1 << std::endl;
        return 0;
    }
    int a0 = n - 1, a1 = n + 2 * k;
    std::vector<int> dp(n + k); // -1: declarer win, 0: declarer lose
    dp.at(n - 1) = -1;
    for (int i = n - 2; i >= 0; i--) {
        int d0 = (a0 - i) / 2, d1 = (a1 - i) / 2, d2 = (k + 1) / 2;
        std::vector<int> w;
        if (a0 <= i + k) w.push_back(a0 - i);
        if ((i + a0) % 2 == 0 and d0 <= k and dp.at(i + d0) == d0) w.push_back(d0);
        if ((i + a1) % 2 == 0 and d1 <= k and dp.at(i + d1) == d1) w.push_back(d1);
        if ((k + 1) % 2 == 0 and dp.at(i + d2) == d2) w.push_back(d2);
        std::sort(w.begin(), w.end());
        decltype(w)::iterator r = std::unique(w.begin(), w.end());
        w.erase(r, w.end());
        if (i == 0) {
            if (w.size() == 0) {
                std::cout << 0 << std::endl;
            } else {
                for (int x : w) {
                    std::cout << x << std::endl;
                }
            }
        } else if (w.size() == 0) {
            dp.at(i) = -1;
            a1 = a0;
            a0 = i;
        } else if (w.size() == 1) {
            dp.at(i) = w.at(0);
        }
    }
}
0