結果

問題 No.181 A↑↑N mod M
ユーザー t98slidert98slider
提出日時 2022-07-20 23:49:33
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,335 bytes
コンパイル時間 1,706 ms
コンパイル使用メモリ 172,184 KB
実行使用メモリ 8,576 KB
最終ジャッジ日時 2024-07-02 14:11:38
合計ジャッジ時間 9,343 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 TLE -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

struct stable_int{
    long long v;
    stable_int() : v(0) {}
    stable_int(long long _v) : v(_v){}
    stable_int& operator++() {
        if(__builtin_add_overflow(v, 1, &v))v = std::numeric_limits<long long>::max();
        return *this;
    }
    stable_int& operator--() {
        if(__builtin_sub_overflow(v, 1, &v))v = std::numeric_limits<long long>::min();
        return *this;
    }
    stable_int operator++(int) {
        stable_int result = *this;
        ++*this;
        return result;
    }
    stable_int operator--(int) {
        stable_int result = *this;
        --*this;
        return result;
    }
    stable_int& operator+=(const stable_int& rhs) {
        if(__builtin_add_overflow(v, rhs.v, &v))v = std::numeric_limits<long long>::max();
        return *this;
    }
    stable_int& operator-=(const stable_int& rhs) {
        if(__builtin_sub_overflow(v, rhs.v, &v))v = std::numeric_limits<long long>::min();
        return *this;
    }
    stable_int& operator*=(const stable_int& rhs) {
        long long pre = v;
        if(__builtin_mul_overflow(v, rhs.v, &v)){
            v = (pre > 0) ^ (rhs.v > 0) ? std::numeric_limits<long long>::min()
                                        : std::numeric_limits<long long>::max();
        }
        return *this;
    }
    stable_int& operator/=(const stable_int& rhs) {
        v /= rhs.v;
        return *this ; 
    }
    stable_int operator+() const { return *this; }
    stable_int operator-() const { return stable_int() - *this; }
    friend stable_int operator+(const stable_int& lhs, const stable_int& rhs) {
        return stable_int(lhs) += rhs;
    }
    friend stable_int operator-(const stable_int& lhs, const stable_int& rhs) {
        return stable_int(lhs) -= rhs;
    }
    friend stable_int operator*(const stable_int& lhs, const stable_int& rhs) {
        return stable_int(lhs) *= rhs;
    }
    friend stable_int operator/(const stable_int& lhs, const stable_int& rhs) {
        return stable_int(lhs) /= rhs;
    }
    friend bool operator==(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v == rhs.v);
    }
    friend bool operator!=(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v != rhs.v);
    }
    friend bool operator<(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v < rhs.v);
    }
    friend bool operator<=(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v <= rhs.v);
    }
    friend bool operator>(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v > rhs.v);
    }
    friend bool operator>=(const stable_int& lhs, const stable_int& rhs) {
        return (lhs.v >= rhs.v);
    }
    friend istream& operator>>(istream& is,stable_int& rhs) noexcept {
        long long _v;
        rhs = stable_int{(is >> _v, _v)};
        return is;
    }
    friend ostream& operator << (ostream &os, const stable_int& rhs) noexcept {
        return os << rhs.v;
    }
};

int main(){
    int A, N, M;
    cin >> A >> N >> M;
    function<int(int, int, int)> rec = [&](int V, int N, int MOD){
        if(N == 0)return 1;
        if(MOD == 1)return 1000000;
        vector<int> order, tb(MOD, -1);
        int val = 1, tim = 0;
        while(tb[val] == -1){
            tb[val] = tim++;
            order.push_back(val);
            val *= V % MOD;
            val %= MOD;
        }
        int EXP = rec(V, N - 1, (tim - tb[val]));
        int idx = (EXP - tb[val]) % (tim - tb[val]) + tb[val];
        return order[idx] + 1000 * MOD;
    };
    function<int(int, int, int)> rec2 = [&](int V, int N, int MOD){
        if(N == 0)return 1;
        if(MOD == 1)return 0;
        vector<int> order, tb(MOD, -1);
        int val = 1, tim = 0;
        while(tb[val] == -1){
            tb[val] = tim++;
            order.push_back(val);
            val *= V % MOD;
            val %= MOD;
        }
        int EXP = rec2(V, N - 1, (tim - tb[val]));
        stable_int v = 1, c = V;
        while(EXP){
            if(EXP & 1)v *= c;
            c *= c;
            EXP >>= 1;
        }
        return int(v.v);
    };
    int ans = rec(A, N, M) % M;
    int ans2 = (N <= 10 ? rec2(A, N, M) : M + 500);
    if(ans2 < M){
        cout << ans2 << endl;
    }else{
        cout << ans << endl;
    }
}
0