玲珑杯1144 - 数论你还会快速幂【结论+快速乘】

1144 - 数论你还会快速幂

Time Limit：5s Memory Limit：256MByte

Submissions：520Solved：114

DESCRIPTION

今天$HH$在学数论,他看到一个很优美的式子:

${\sum }_{i=1}^n{i}^{k}modp$

一向热衷于抱队友大腿的$HH$便问队友$ZZ$怎么做

$ZZ$:"$n,k$多大?"

$HH$:"${10}^5,{10}^5$"

$ZZ$:"快速幂嘛"

$HH$:"${10}^9,{10}^5$"

$ZZ$:"拉格朗日插值嘛"

$HH$:"${10}^{18},{10}^{18}$"

队友:"让我想想.."

INPUT

第一行是一个整数 $T(1\le T\le 1000)$,表示有 $T$组数据对于每组数据输入一行3个整数 $n,k,p(1\le n\le {10}^{18},0\le k\le {10}^{18},1\le p\le {10}^{18},0\le n-p\le 100)$且模数 $p$ 是质数,

OUTPUT

对于每组数据输出题目中的表达式的值

SAMPLE INPUT

25 2 37 2 3

SAMPLE OUTPUT

12

HINT

对于第一组样例我们有 $1^2+2^2+3^2+4^2+5^2=1\left(mod3\right)$对于第二组样例我们有 $1^2+2^2+3^2+4^2+5^2+6^2+7^2=2\left(mod3\right)$

题解结论：（n-p肯定是解题的关键）k如果是p-1的倍数，前p个数的和是p-1，如果不是，前p个数的和是0。后面只需要计算n%p的结果了，最多100次，暴力解决，由于n,k,p，比较大，采用快速乘代替快速幂；

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
ll n,k,p;

ll qsum(ll a,ll b)  //快速乘防止爆 LL 
{
    ll ans=0;
    while(b)
    {
        if(b&1) ans=(ans+a)%p;
        a=(a+a)%p;
        b/=2;
    }
    return ans;
}

ll qpow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=(qsum(ans,a))%p;
        b>>=1;
        a=qsum(a,a)%p;
    }
    return ans;
}

int main()
{
    int t;scanf("%d",&t);
    while(t--)
    {
        scanf("%lld%lld%lld",&n,&k,&p);
        if(k==0)
        {
            printf("%lld\n",n%p);
            continue;
        }
        ll sum=0;
        if(k%(p-1)==0) //结论，可以选择记忆 
        {
        	sum=qsum(n/p,p-1)%p;
		}
        n%=p;
        for(ll i=1;i<=n;i++)
            sum=(sum+qpow(i,k))%p;
        printf("%lld\n",(sum+p)%p);
    }
    return 0;
}