結果

問題 No.435 占い(Extra)
コンテスト
ユーザー himueu
提出日時 2026-10-20 02:56:06
言語 C++23(gcc16)
(gcc 16.2.0 + boost 1.92.0 + ACL)
コンパイル:
g++-16 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
MLE  
実行時間 -
コード長 1,606 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,202 ms
コンパイル使用メモリ 174,988 KB
実行使用メモリ 159,952 KB
最終ジャッジ日時 2026-10-20 02:57:03
合計ジャッジ時間 7,457 ms
ジャッジサーバーID
(参考情報)
judge1_0 / judge6_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 28 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include <iostream>
#include <vector>

void solve() {
    using namespace std;
    int n, x, a, b, m;
    cin >> n >> x >> a >> b >> m;
    n--;

    vector<int> fac_f(n + 1), fac_e(n + 1);
    fac_f[0] = 1;
    fac_e[0] = 0;
    for (int i = 0; i < n; i++) {
        int k = i + 1, e = 0;
        while (k % 3 == 0) {
            k /= 3;
            e++;
        }
        fac_f[i + 1] = fac_f[i] * k % 9;
        fac_e[i + 1] = fac_e[i] + e;
    }
    constexpr int inv[] = {0, 1, 5, 0, 7, 2, 0, 4, 8};
    vector<int> ifac_f(n + 1), ifac_e(n + 1);
    ifac_f[n] = inv[fac_f[n]];
    ifac_e[n] = -fac_e[n];
    for (int i = n; i > 0; i--) {
        int k = i, e = 0;
        while (k % 3 == 0) {
            k /= 3;
            e++;
        }
        ifac_f[i - 1] = ifac_f[i] * k % 9;
        ifac_e[i - 1] = ifac_e[i] + e;
    }
    auto comb_mod9 = [&](int n, int r) -> int {
        if (n < 0 || r < 0 || n < r) { return 0; }
        int res = fac_f[n] * ifac_f[r] * ifac_f[n - r] % 9;
        int e = fac_e[n] + ifac_e[r] + ifac_e[n - r];
        if (e > 1) { return 0; }
        if (e == 1) { res = res * 3 % 9; }
        return res;
    };

    bool all_zero = true;
    int ans = 0;
    for (int i = 0; i <= n; i++) {
        int c = x % 10;
        x ^= a;
        x += b;
        x %= m;
        all_zero &= (c == 0);
        ans += (c - 0) * comb_mod9(n, i);
        ans %= 9;
    }
    if (!all_zero && ans == 0) { ans = 9; }
    cout << ans << "\n";
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int t;
    std::cin >> t;
    while (t--) { solve(); }
}
0