立志用最少的代碼做最高效的表達(dá)

---

It’s easy for ACMer to calculate A^X mod P. Now given seven integers n, A, K, a, b, m, P, and a function f(x) which defined as following.
f(x) = K, x = 1
f(x) = (a*f(x-1) + b)%m , x > 1
Now, Your task is to calculate
$(A^{f(1)}+A^{f(2)}+A^{f(3)}+......+A^{f(n)})modularP.$

Input
In the first line there is an integer T (1 < T <= 40), which indicates the number of test cases, and then T test cases follow. A test case contains seven integers n, A, K, a, b, m, P in one line.
1 <= n <= 10^6
0 <= A, K, a, b <= 10^9
1 <= m, P <= 10^9

Output
For each case, the output format is “Case #c: ans”.
c is the case number start from 1.
ans is the answer of this problem.

Sample Input
2
3 2 1 1 1 100 100
3 15 123 2 3 1000 107
Sample Output
Case #1: 14
Case #2: 63

---

分析

1e9的規(guī)模，即使O(n)的復(fù)雜度也沒(méi)辦法通過(guò)，因此考慮將指數(shù)分解

把指數(shù) f(x) 分解為 $f(x)=ak+b$，這樣一來(lái)， $A^{f(x)}=A^{ak}?A^b$。

上式中的 k 和要打的表的大小有關(guān)，哪表開(kāi)多大合適呢？理論上開(kāi) 33333 就可以了，因?yàn)?sqrt(1e9) = 33333。這樣只要計(jì)算 $(A^1)\%P、(A^2)\%P\dots \dots (A^{33333})\%P$ 以及 $(A^{1?33333})\%P(A^{2?33333})\%P\dots \dots (A^{33333?33333})\%P$ ，對(duì)于 $A^{f(x)}$，我們就可以通過(guò)上面兩個(gè)數(shù)組相乘的結(jié)果得到了。

注意：

由于代碼中涉及到乘法運(yùn)算，所以 int 型會(huì)溢出，可以全部開(kāi) long long。

---

#include<bits/stdc++.h> #define len 35000 using namespace std;typedef long long gg; gg n, A, K, a, b, m, P; gg dp1[len + 10], dp2[len + 10]; void init() {gg i;dp1[0] = dp2[0] = 1;for(i = 1; i <= len; i++) dp1[i] = (dp1[i-1]*A)%P;dp2[1] = dp1[len];for(i = 2; i <= len; i++)dp2[i] = (dp2[i-1]*dp2[1])%P; }int main() {gg T; cin >> T; for(gg j = 1; j <= T; j++){cin >> n >> A >> K >> a >> b >> m >> P;init();gg ans = 0;for(gg i = 1; i <= n; i++) {ans = (ans + (dp1[K%len]*dp2[K/len])%P)%P;K = (a * K + b) % m;}printf("Case #%lld: %lld\n", j, ans);}return 0; }

總結(jié)

以上是生活随笔為你收集整理的【简单数论】H - A^X mod P_HRBUST - 2049_31行代码AC的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。