結果

問題 No.181 A↑↑N mod M
ユーザー yosupotyosupot
提出日時 2015-04-06 00:19:50
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 2,203 bytes
コンパイル時間 1,268 ms
コンパイル使用メモリ 168,352 KB
実行使用メモリ 13,880 KB
最終ジャッジ日時 2024-07-04 02:22:37
合計ジャッジ時間 7,690 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
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;
typedef unsigned int uint;
typedef long long ll;
typedef unsigned long long ull;

typedef pair<ll, ll> P;

// x^n % md
ll pow_mod(ll x, ll n, P md) {
    ll y = 1;
    for (int i = 0; i < n; i++) {
        y *= x;
        if (y > md.first) {
            y -= md.first;
            y %= md.second;
            y += md.first;
        }
    }
    return y;
/*    ll r = 1;
    while (n) {
        if (n & 1) {
            r = r * x;
            if (r > md.first) {
                r -= md.first;
                r %= md.second;
                r += md.first;
            }
        }
        x = x * x;
        if (x > md.first) {
            x -= md.first;
            x %= md.second;
            x += md.first;
        }
        n >>= 1;
    }
    return r;*/
}


// x^n % md
ll pow_mod(ll x, ll n, ll md) {
    ll r = 1;
    while (n) {
        if (n & 1) r = (r * x) % md;
        x = (x * x) % md;
        n >>= 1;
    }
    return r % md;
}


P cycle(ll a, ll m) {
    ll x = 1;
    map<ll, ll> mp;
    mp[x] = 0;
    for (int i = 1; i < 100000; i++) {
        x *= a;
        x %= m;
        if (mp.count(x)) {
            return {mp[x], i-mp[x]};
        }
        mp[x] = i;
    }
    assert(false);
}

int main() {
    ll a, n, m;
    cin >> a >> n >> m;
    if (a == 1 || n == 0) {
        cout << 1 % m << endl;
        return 0;
    }
    if (m == 1) {
        cout << 0 << endl;
        return 0;
    }
    n--;
    P p = cycle(a, m);
    assert(p.second);
    ll x = 1;
    if (a < 10000) {
        for (int i = 0; i < n; i++) {
            x = pow_mod(a, x, p);
        }
        cout << pow_mod(a, x, m) << endl;
        return 0;
    }
    assert(false);
    map<ll, ll> mp;
    mp[x] = 0;
    bool f = false;
    for (int i = 1; i < 100000; i++) {
        x = pow_mod(a, x, p);
        if (mp.count(x)) {
            f = true;
            if (n < mp[x]) break;
            n -= mp[x];
            n %= i - mp[x];
            n += mp[x];
            break;
        }
        mp[x] = i;
    }
    assert(f);
    x = 1;
    for (int i = 0; i < n; i++) {
        x = pow_mod(a, x, p);
    }
    cout << pow_mod(a, x, m) << endl;
    return 0;
}
0