結果

問題 No.1770 N言っちゃダメゲーム (6)
ユーザー hitonanode
提出日時 2021-12-02 01:12:41
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,125 bytes
コンパイル時間 1,100 ms
コンパイル使用メモリ 93,136 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-05 01:45:34
合計ジャッジ時間 2,370 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC optimize("O3,unroll-loops")
#include <iostream>
#include <vector>
using namespace std;

int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);
    int N, K;
    cin >> N >> K;

    constexpr int All = -1, None = -2;
    vector<int> can_be_zero(N, All);
    for (int n = N - 1; n > 0; --n) {
        if (can_be_zero[n] == All) {
            for (int i = 1; i <= K; ++i) {
                if (n - i < 0) break;
                if (can_be_zero[n - i] == All) can_be_zero[n - i] = i;
                else {
                    can_be_zero[n - i] = None;
                }
            }
        } else if (can_be_zero[n] != None) {
            int i = can_be_zero[n];
            if (n - i >= 0) {
                if (can_be_zero[n - i] == All) can_be_zero[n - i] = i;
                else {
                    can_be_zero[n - i] = None;
                }
            }
        }
    }

    vector<int> ret;
    for (int i = 1; i <= K; ++i) {
        if (can_be_zero[i] == i or can_be_zero[i] == All) ret.push_back(i);
    }
    if (ret.empty()) ret.push_back(0);
    for (auto x : ret) cout << x << '\n';
}
0