結果

問題 No.181 A↑↑N mod M
ユーザー koyumeishi
提出日時 2015-04-06 01:52:48
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,500 bytes
コンパイル時間 784 ms
コンパイル使用メモリ 85,084 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-04 03:11:34
合計ジャッジ時間 2,070 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 10 WA * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;

long long gcd(long long a, long long b){
	if(b==0) return a;
	return gcd(b, a%b);
}


long long tot(long long n){
	long long ret = n;
	for(long long i=2; i*i<=n; i++){
		if(n%i == 0){
			ret = ret/i * (i-1);
			while(n%i == 0){
				n /= i;
			}
		}
	}
	if(n != 1){
		ret = ret/n * (n-1);
	}
	return ret;
}

long long mypow(long long x, long long y, long long MOD){
	if(x==0 && y!=0) return 0;
	long long ret=1LL;
	while(y>0LL){
		if(y&1LL) ret = (ret * x) % MOD;
		x = (x*x) % MOD;
		y >>= 1LL;
	}
	return ret;
}

vector<pair<long long, long long> > prime_factorization(long long N){
	vector< pair<long long, long long> > ret;
	
	for(long long i=2; i*i<=N; i++){
		long long tmp = 0;
		while(N%i==0){
			tmp++;
			N/=i;
		}
		if(tmp>0){
			ret.push_back( make_pair(i, tmp) );
		}
	}
	if(N!=1){
		ret.push_back( make_pair(N, 1) );
	}
	return ret;
}

int main(){
	long long A,N,M;
	cin >> A >> N >> M;

	if(N==0){
		cout << 1 << endl;
		return 0;
	}

	long long tot_ = tot(M);

	map<long long, long long> memo;

	long long ret = A;
	memo[ret] = 0;
	for(int i=1; i<N; i++){
		if( gcd(A, M) == 1 ){
			ret %= tot_;
		}
		ret = mypow(A, ret, M);

		if(memo.find(ret) != memo.end()){
			int loop = i - memo[ret];
			i = N - (N-i)%loop;
		}else{
			memo[ret] = i;
		}
	}

	cout << ret << endl;

	return 0;
}
0