玲珑杯 1144 - 数论你还会快速幂（打表找规律）

传送门

解题思路：

官方题解：

B. 数论你还会快速幂

看到$n$比$p$大了不到100个,我们可以思考一下假如我们只算到$p$或者$p-1$会是怎么样的

即考虑函数$\sum_{i=1}^{p-1}i^k\ mod \ p$

我们先打个表能发现当$k$是$p-1$的倍数的时候这个值是$p-1$,别的时候都是$0$

然而这是否正确的呢? 我们可以简单的推导一下:

假设$g$是$p$的原根,那么根据原根的性质,$g$的1次方到$p-1$次方在模$p$的时候

对应了$1$到$p-1$的某个全排列,即原式等价于

$g^{k} + g^{2k} + g^{3k} +... + g^{(p-1)k}$

然后我们用等比数列求和一下则得到

$(gk−gpk)/(1−gk)=(gk−gpk)×inv(1−gk)$

再根据费马小定理我们便得到了$g^p=g$,那么这个式子就变成了0

不过这是建立在这是等比数列的情况下的,所以我们对几种不是等比数列的情况需要特判:

(1) p为2的时候原根是1

(2) k为p-1的时候由费马小定理可以得到所有项均为1

所以我们现在只需要算剩下的100个左右的数就可以了,快速幂搞一下就行.

不过要小心这个范围乘法会爆long long 所以需要快速乘/O(1)快速乘

自己的解法：

就是通过打表找到规律，在 $[1,p-1]$ 区间中当 $k$ 是 $p-1$ 的倍数的时候值为 $p-1$ ,否则为 $0$， 然后就可以计算了，还需要知道的是 ：$(i+p)^p\%p \ =\ i^p\%p$ ,然后计算之后就只剩下 $100$ 个数了，直接快速幂就OK了，需要特判一下 $k==0$ 的情况。
trick：O(1) 的快速乘法是真的快呀。

代码：

#include <iostream>
#include <math.h>
#include <stdio.h>
using namespace std;

typedef long long LL;
const double eps = 1e-8;
inline LL Mul(LL x, LL y, LL MOD){
    x %= MOD, y %= MOD;
    long long tmp=(x*y-(long long)((long double)x/MOD*y+eps)*MOD);
    return tmp<0 ? tmp+MOD : tmp;
}
LL Pow(LL a, LL b, LL MOD){
    LL ans = 1;
    a %= MOD;
    while(b){
        if(b & 1) ans = Mul(ans, a, MOD);
        b >>= 1;
        a = Mul(a, a, MOD);
    }
    return ans;
}
int main(){
    LL n, k, p;
    int T; scanf("%d", &T);
    while(T--){
        scanf("%lld%lld%lld",&n,&k,&p);
        if(k == 0) {printf("%lld\n", n%p); continue; }
        LL ans = 0;
        if(k%(p-1) == 0) ans = Mul(n/p, p-1, p);
        n %= p;
        for(LL i=1; i<=n; i++) ans = ( ans + Pow(i, k, p) ) % p;
        printf("%lld\n",ans);
    }
    return 0;
}