結果

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

ソースコード

diff # 

/*
 * validator
 */

#include <bits/stdc++.h>

std::pair<int, char> read() {
    int x = 0;
    char c;
    while (true) {
        c = std::getchar();
        if (!std::isdigit(c)) {
            return std::pair(x, c);
        }
        x = 10 * x + c - '0';
    }
}

int main() {
    char c;
    int n, k;
    std::tie(n, c) = read();
    assert(c == ' ');
    std::tie(k, c) = read();
    assert(c == '\n');
    assert(std::getchar() == EOF);

    if (k==1) {
        std::cout << 1 << '\n';
        return 0;
    }

    std::vector<std::vector<int>> a(n+1);
    std::vector<int> b = {1};
    for (int i = 2; i <= n; i++) {
        auto&& v = a.at(i);
        int x = b.back();
        auto transite = [&](int j) {
            int d = i - j;
            if (d <= k && (a.at(j).empty()
                       || (a.at(j).size() == int{1} && a.at(j).front() == d)))
            {
                v.push_back(d);
            }
        };
        transite(x);
        transite((i + x) / 2);
        if (int{1} != b.size()) {
            int y = b.at(b.size() - 2);
            transite(y + k + 1);
            transite((i + y) / 2);
        }
        std::sort(v.begin(), v.end());
        v.resize(std::unique(v.begin(), v.end()) - v.begin()); if (v.empty()) b.push_back(i);
    }

    std::vector<int> ans = a.at(n);
    if (ans.empty()) {
        std::cout << 0 << '\n';
    } else {
        for (int x: ans) {
            std::cout << x << '\n';
        }
    }
}
0