Submission #39989476

Source Code Expand

#include<iostream>
#include<vector>
#include<cmath>

using namespace std;

typedef long long ll;

ll A, X, M;

int main() {
	cin >> A >> X >> M;

	ll ans = 0;
	ll rtX = sqrt(X);
	ll sum0toRtX = 0;
	ll lastVal = 0;//毎回のループでスケーリングする倍数
	ll multipliedLastVal = 0;//↑の変数をループによって実際にスケーリングさせたもの。
	ll lastElementsSum = 0;
	
	{
		ll now = 1;
		lastVal = now;
		for (ll i = 0; i < rtX; i++) {
			sum0toRtX += now;
			sum0toRtX %= M;
			if ((X % rtX) > i) {
				lastElementsSum += now;
				lastElementsSum %= M;
			}
			now *= A;
			now %= M;
			lastVal = now;
		}
		ans += sum0toRtX;
		ans %= M;
		multipliedLastVal = lastVal;
	}
	
	{
		for (ll i = 1; (i + 1) * rtX <= X; i++) {
			sum0toRtX *= lastVal;
			sum0toRtX %= M;
			ans += sum0toRtX;
			ans %= M;
			multipliedLastVal *= lastVal;
			multipliedLastVal %= M;
		}
	}

	{
		lastElementsSum *= multipliedLastVal;
		lastElementsSum %= M;
		ans += lastElementsSum;
		ans %= M;
	}

	cout << ans << endl;

	return 0;
}

Submission Info

Submission Time
Task E - Geometric Progression
User Sen
Language C++ (GCC 9.2.1)
Score 500
Code Size 1107 Byte
Status AC
Exec Time 60 ms
Memory 3632 KiB

Judge Result

Set Name Sample All AfterContest
Score / Max Score 0 / 0 500 / 500 0 / 0
Status
AC × 3
AC × 31
AC × 1
Set Name Test Cases
Sample sample00.txt, sample01.txt, sample02.txt
All sample00.txt, sample01.txt, sample02.txt, testcase00.txt, testcase01.txt, testcase02.txt, testcase03.txt, testcase04.txt, testcase05.txt, testcase06.txt, testcase07.txt, testcase08.txt, testcase09.txt, testcase10.txt, testcase11.txt, testcase12.txt, testcase13.txt, testcase14.txt, testcase15.txt, testcase16.txt, testcase17.txt, testcase18.txt, testcase19.txt, testcase20.txt, testcase21.txt, testcase22.txt, testcase23.txt, testcase24.txt, testcase25.txt, testcase26.txt, testcase27.txt
AfterContest after_contest_01.txt
Case Name Status Exec Time Memory
after_contest_01.txt AC 10 ms 3552 KiB
sample00.txt AC 2 ms 3464 KiB
sample01.txt AC 2 ms 3492 KiB
sample02.txt AC 46 ms 3496 KiB
testcase00.txt AC 47 ms 3500 KiB
testcase01.txt AC 50 ms 3548 KiB
testcase02.txt AC 2 ms 3540 KiB
testcase03.txt AC 2 ms 3468 KiB
testcase04.txt AC 49 ms 3488 KiB
testcase05.txt AC 50 ms 3452 KiB
testcase06.txt AC 2 ms 3632 KiB
testcase07.txt AC 2 ms 3504 KiB
testcase08.txt AC 60 ms 3356 KiB
testcase09.txt AC 47 ms 3464 KiB
testcase10.txt AC 48 ms 3632 KiB
testcase11.txt AC 22 ms 3492 KiB
testcase12.txt AC 30 ms 3428 KiB
testcase13.txt AC 27 ms 3492 KiB
testcase14.txt AC 42 ms 3544 KiB
testcase15.txt AC 30 ms 3504 KiB
testcase16.txt AC 43 ms 3424 KiB
testcase17.txt AC 49 ms 3500 KiB
testcase18.txt AC 52 ms 3444 KiB
testcase19.txt AC 36 ms 3488 KiB
testcase20.txt AC 45 ms 3504 KiB
testcase21.txt AC 46 ms 3448 KiB
testcase22.txt AC 26 ms 3540 KiB
testcase23.txt AC 51 ms 3488 KiB
testcase24.txt AC 36 ms 3540 KiB
testcase25.txt AC 48 ms 3488 KiB
testcase26.txt AC 46 ms 3552 KiB
testcase27.txt AC 9 ms 3424 KiB