結果

問題 No.181 A↑↑N mod M
ユーザー pyranine
提出日時 2020-12-23 22:39:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 967 bytes
コンパイル時間 1,774 ms
コンパイル使用メモリ 173,372 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-21 16:35:15
合計ジャッジ時間 3,369 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 6
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#define MOD(a,mod) ((a)<(mod)?(a):(a)%(mod)+(mod))

int power(long long n, long long k, const int mod) {
  long long ans = MOD(1,mod);
  while (k) {
    if (k&1) ans = MOD(ans*n,mod);
    n = MOD(n*n,mod);
    k >>= 1;
  }
  return ans;
}

unordered_map<int,int> mp;
int totient(int n) {
  if (mp.find(n) != mp.end()) {
    return mp[n];
  }
  int ans = n;
  for (int i = 2; i * i <= n; i++) {
    if(n % i == 0) {
      while (n % i == 0) n /= i;
      ans = ans / i * (i - 1);
    }
  }
  if (n > 1) ans = ans / n * (n - 1);
  return mp[n] = ans;
}

int tetration(int a, int b, int mod) {
  if (a == 0) return MOD((b + 1) % 2, mod);
  if (b == 0) return MOD(1, mod);
  if (mod == 1) return MOD(1, mod);
  return power(a, tetration(a, b - 1, totient(mod)), mod);
}

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  int a, b, m;
  cin >> a >> b >> m;
  cout << tetration(a, b, m) % m << endl;

  return 0;
}
0