#141927 (C++14) No.181 A↑↑N mod M

提出ソース

結果

問題	No.181 A↑↑N mod M
ユーザー	pekempey
提出日時	2016-12-28 15:21:55
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	3 ms / 5,000 ms
コード長	2,176 bytes
コンパイル時間	1,707 ms
コンパイル使用メモリ	166,940 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-05-08 23:47:47
合計ジャッジ時間	3,033 ms
ジャッジサーバーID （参考情報）	judge1 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 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	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	3 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	3 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	AC	2 ms 5,376 KB
testcase_28	AC	2 ms 5,376 KB
testcase_29	AC	2 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	2 ms 5,376 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	2 ms 5,376 KB
testcase_35	AC	2 ms 5,376 KB
testcase_36	AC	2 ms 5,376 KB
testcase_37	AC	2 ms 5,376 KB
testcase_38	AC	2 ms 5,376 KB
testcase_39	AC	2 ms 5,376 KB
testcase_40	AC	1 ms 5,376 KB
testcase_41	AC	2 ms 5,376 KB
testcase_42	AC	3 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

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

int phi[2001];

int modpow(int a, int b, int m) {
	int ret = 1;
	while (b > 0) {
		if (b & 1) ret = 1LL * ret * a % m;
		a = 1LL * a * a % m;
		b /= 2;
	}
	return ret;
}

int lessThanM(int a, int n, int m) {
	if (a == 1 || n == 0) return 1 < m;
	if (n == 1) {
		return a < m;
	} else if (n == 2) {
		if (a == 2) return 4 < m;
		if (a == 3) return 27 < m;
		if (a == 4) return 256 < m;
	} else if (n == 3) {
		if (a == 2) return 16 < m;
	}
	return false;
}

// [theorem] a^k mod m はphi[m]の周期を持つ（ただし周期に入るまでに最大でphi[m]の時間がかかる)
// [proof] 
// gcd(p,m)=1 となるように a=pq と分解する。
// このとき p^k が周期 phi(m) となることはオイラーの定理として知られている。
// q^k mod mを考えよう。
// まず m=XY と分解する。
// X,Yはどうなっているのかというと、qが持つ素因数をそのままXに持ってきて、それ以外をYに持ってきている。
// このとき q^k mod X は k>=phi(X) で 0 になる。
// 一方、gcd(q,Y)=1なので、q^k mod Yは周期 phi(Y) を持つ。
// 中国剰余定理により、q^k mod XY は周期 phi(Y) を持つことが示された。
// p^k mod XY が周期 phi(XY) を持ち、q^k mod XY が周期 phi(Y) を持つので、
// (pq)^k mod XY は周期 phi(XY) を持つ。（lcm(phi(X), phi(XY))=phi(XY)。
//
// a^^(n-1)<phi[m]の場合は周期に入っていない可能性があるので、そのまま計算する。
// a^^(n-1)>=phi[m]の場合はすでに周期に入っているので、phi[m]を足しておくとちょうど良い。
int tetra(int a, int n, int m) {
	if (n == 0) return 1 % m;
	if (m == 1) return 0;
	if (lessThanM(a, n - 1, phi[m])) {
		return modpow(a, tetra(a, n - 1, phi[m]), m);
	} else {
		return modpow(a, tetra(a, n - 1, phi[m]) + phi[m], m);
	}
}

int main() {
	for (int i = 0; i <= 2000; i++) {
		phi[i] = i;
	}
	for (int i = 2; i <= 2000; i++) {
		if (phi[i] == i) {
			for (int j = i; j <= 2000; j += i) {
				phi[j] = phi[j] / i * (i - 1);
			}
		}
	}

	int a, n, m;
	cin >> a >> n >> m;
	cout << tetra(a, n, m) << endl;
}

yukicoder

結果

テストケース

ソースコード