#531031 (C++17) No.1161 Many Powers

• 問題: No.1161 Many Powers
• ユーザー: ningenMe
• 提出日時: 2020-08-15 06:24:22
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 75 ms / 2,000 ms
• コード長: 736 bytes
• コンパイル時間: 2,053 ms
• コンパイル使用メモリ: 201,828 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-10 17:12:13
• 合計ジャッジ時間: 3,459 ms
• ジャッジサーバーID （参考情報）: judge3 / 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,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	5 ms; 5,248 KB
testcase_06	AC	4 ms; 5,248 KB
testcase_07	AC	1 ms; 5,248 KB
testcase_08	AC	31 ms; 5,248 KB
testcase_09	AC	35 ms; 5,248 KB
testcase_10	AC	14 ms; 5,248 KB
testcase_11	AC	25 ms; 5,248 KB
testcase_12	AC	13 ms; 5,248 KB
testcase_13	AC	41 ms; 5,248 KB
testcase_14	AC	62 ms; 5,248 KB
testcase_15	AC	52 ms; 5,248 KB
testcase_16	AC	69 ms; 5,248 KB
testcase_17	AC	46 ms; 5,248 KB
testcase_18	AC	34 ms; 5,248 KB
testcase_19	AC	20 ms; 5,248 KB
testcase_20	AC	73 ms; 5,248 KB
testcase_21	AC	20 ms; 5,248 KB
testcase_22	AC	75 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	AC	2 ms; 5,248 KB

ソースコード

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

class ArbitraryMod{
public:
	//Pow_Mod O(log(n))
	inline static long long pow(long long x, long long n, long long mod) {
		long long res = 1;
		for (x %= mod; n > 0; n >>= 1, (x *= x) %= mod) if (n & 1) (res *= x) %= mod;
		return res;
	}
	//Inv_Mod O(log(mod))
	inline static long long inv(long long x, long long mod){
		return pow(x,mod-2,mod); 
	}
};

int main() {
	long long A,B,C; cin >> A >> B >> C;
	long long ans = 0;
	vector<long long> cnt(C+1,0);
	for(long long i = 1; i <= C; ++i) {
		cnt[i] = ArbitraryMod::pow(i,B,C);
	}
	for(long long i = 1; i <= A%C; ++i) {
		ans += cnt[i];
	}
	ans += (accumulate(cnt.begin(),cnt.end(),0LL)%C)*(A/C);
	cout << ans%C << endl;
    return 0;
}