結果

問題 No.435 占い(Extra)
ユーザー pekempey
提出日時 2016-11-22 19:11:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 127 ms / 2,000 ms
コード長 1,899 bytes
コンパイル時間 5,364 ms
コンパイル使用メモリ 181,152 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-08 12:13:20
合計ジャッジ時間 5,824 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include <bits/stdc++.h>
using namespace std;

#ifndef LOCAL
#define getchar getchar_unlocked
#define putchar putchar_unlocked
#endif

int in() {
    int n, c;
    while ((c = getchar()) < '0') if (c == EOF) abort();
    n = c - '0';
    while ((c = getchar()) >= '0') n = n * 10 + c - '0';
    return n;
}

void out(int n) {
    int res[11], i = 0;
    do { res[i++] = n % 10, n /= 10; } while (n);
    while (i) putchar(res[--i] + '0');
    putchar('\n');
}

char p2[256], inv[9];

pair<char, int> decomp(int n) {
    if (n == 0) return{ 0, 0 };
    int three = 0;
    while (n % 3 == 0) {
        n /= 3;
        three++;
    }
    return{ inv[n % 9], three };
}

void upd(char &x, char y) {
    if ((x += y) >= 6) x -= 6;
}

int main() {
    p2[0] = 1;
    for (int i = 1; i < 256; i++) p2[i] = p2[i - 1] * 2 % 9;
    for (int i = 0; i < 6; i++) inv[p2[i]] = i;

    int T = in();
    while (T--) {
        int n = in() - 1, x = in(), a = in(), b = in(), m = in();
        auto N = decomp(n);
        pair<char, int> L(0, 0);
        pair<char, int> R = N;
        bool zero = true;
        uint32_t ans = 0;
        for (int i = 0; i <= n; i++) {
            if (zero && x % 10 != 0) zero = false;
            char two = N.first + (6 - L.first) + (6 - R.first);
            int three = N.second - L.second - R.second;
            if (three <= 1) {
                int c = p2[two] * (three == 0 ? 1 : three == 1 ? 3 : 0);
                ans += c * (x % 10);
            }
            auto ll = decomp(i + 1);
            auto rr = decomp(n - i);
            upd(L.first, ll.first);
            L.second += ll.second;
            upd(R.first, 6 - rr.first);
            R.second -= rr.second;
            x = ((x ^ a) + b) % m;
        }
        ans %= 9;
        if (ans == 0) ans = 9;
        if (zero) ans = 0;
        out(ans);
    }
}
0