結果

問題 No.434 占い
コンテスト
ユーザー himueu
提出日時 2026-10-20 02:46:54
言語 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
結果
AC  
実行時間 14 ms / 2,000 ms
+ 507µs
コード長 1,532 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,843 ms
コンパイル使用メモリ 174,500 KB
実行使用メモリ 9,796 KB
最終ジャッジ日時 2026-10-20 02:47:08
合計ジャッジ時間 4,963 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>

void solve() {
    using namespace std;
    string s;
    cin >> s;
    int n = s.size() - 1;

    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++) {
        all_zero &= (s[i] == '0');
        ans += (s[i] - '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