結果

問題 No.435 占い(Extra)
コンテスト
ユーザー kimiyuki
提出日時 2016-10-15 00:40:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 473 ms / 2,000 ms
コード長 2,062 bytes
コンパイル時間 1,785 ms
コンパイル使用メモリ 78,224 KB
実行使用メモリ 42,312 KB
最終ジャッジ日時 2024-10-08 11:56:20
合計ジャッジ時間 6,146 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <tuple>
#include <cassert>
#define repeat(i,n) for (int i = 0; (i) < (n); ++(i))
#define repeat_from(i,m,n) for (int i = (m); (i) < (n); ++(i))
#define whole(f,x,...) ([&](decltype((x)) y) { return (f)(begin(y), end(y), ## __VA_ARGS__); })(x)
using namespace std;
template <class T> void setmax(T & a, T const & b) { if (a < b) a = b; }
template <typename T, typename X> auto vectors(T a, X x) { return vector<T>(x, a); }
template <typename T, typename X, typename Y, typename... Zs> auto vectors(T a, X x, Y y, Zs... zs) { auto cont = vectors(a, y, zs...); return vector<decltype(cont)>(x, cont); }
pair<int,int> zero(int k) {
    int z = 0;
    while (k % 3 == 0) { k /= 3; z += 1; }
    return { k, z };
}
int pack(int k, int z) {
    return z * 9 + k;
}
pair<int,int> unpack(int n) {
    return { n % 9, n / 9 };
}
const int inv[9] = { 0, 1, 5, 0, 7, 2, 0, 4, 8 };
int choose(int n, int r) {
    static vector<int> fact(1, pack(1, 0));
    if (fact.size() <= n) {
        int l = fact.size();
        fact.resize( n + 1);
        repeat_from (i,l,n+1) {
            int pk, pz; tie(pk, pz) = unpack(fact[i-1]);
            int k, z; tie(k, z) = zero(i);
            fact[i] = pack(pk * k % 9, pz + z);
        }
    }
    r = min(r, n - r);
    int pk, pz; tie(pk, pz) = unpack(fact[n]);
    int qk, qz; tie(qk, qz) = unpack(fact[r]);
    int rk, rz; tie(rk, rz) = unpack(fact[n-r]);
    int p = pk * inv[qk] * inv[rk] % 9;
    int z = pz - qz - rz;
    assert (z >= 0);
    return z == 0 ? p : z == 1 ? p * 3 % 9 : 0;
}
int main() {
    int t; cin >> t;
    while (t --) {
        int n, x, a, b, m; cin >> n >> x >> a >> b >> m;
        bool zero = true;
        int acc = x % 10;
        if (x % 10 != 0) zero = false;
        repeat_from (i,1,n) {
            x = ((x ^ a) + b) % m;
            acc = (acc + x % 10 * choose(n-1, i)) % 9;
            if (x % 10 != 0) zero = false;
        }
        if (not zero and acc == 0) acc = 9;
        cout << acc << endl;
    }
    return 0;
}
0