2019牛客暑期多校训练营（第九场）A.The power of Fibonacci(循环节)

题意

$$\{\begin{array}{c}{f}_0=0\\ f_1=1\\ f_n=f_{n-1}+f_{n-2},n\ge 2\end{array}$$
给你$n$和$m$求
$$\sum _{i=0}^n{f}_i^m\%1000000000$$
$1\le n\le 10^9,1\le m\le 1000$

思路

Fibonomial coefficients+常系数线性递推
由于在$mod=1000000000$下$5$的二次剩余不存在，所以不能变为通项的那种方法来做。

那我们可以换一种思路，易知在模意义下斐波那契数列的循环节和次幂的循环节是一样的，原数是循环的那这个数的次幂也是循环的。但是$mod$太大了，也可以直接去找循环节，直接去找的循环节为1500000000，对于计算答案来说没有什么贡献。但有

$mod=1000000000=2^9\ast 5^9=512\ast 1953125$

我们可以求出在模$512$下的$a_1={\sum }_{i=0}^n{f}_i^m\%512$和在模$1953125$下$a_2={\sum }_{i=0}^n{f}_i^m\%1953125$，那么我们就可以得到同余方程组
$$\{\begin{array}{c}x\equiv a_1\mathrm{m}\mathrm{o}\mathrm{d}512\\ x\equiv a_2\mathrm{m}\mathrm{o}\mathrm{d}1953125\end{array}$$
对于这个同余方程组我们套一个中国剩余定理$(CRT)$就可以得到最终$\mathrm{m}\mathrm{o}\mathrm{d}1000000000$的答案了
下面我再说一下怎么求解$a_1$和$a_2$

我们先找一下斐波那契数列对于$512$和$1953125$的循环节，由于这两个数都不大，直接用for循环就可以找到，我们得到的循环节分别是$loo{p}_1=768$,$loo{p}_2=7812500$
我们存下在模$512$和$1953125$下的斐波那契数列的$m$次幂的和，由于存在循环节所以，斐波那契数列的$m$次幂的和也是同样在$768$,$7812500$循环的，所以我们前缀和只用存到$768$,$7812500$就够了记为$sum1[768],sum2[7812500]$。

对于给你的$n$，由于前缀和在$768$,$7812500$是重复的，那么答案由两部分构成一部分是重复完整的循环，另一部分就是最后一段多出来的部分，即
$$a_1=(\frac{n}{768}\ast sum1[768]+sum1[n\%768])\%mod$$
$$a_2=(\frac{n}{7812500}\ast sum1[7812500]+sum1[n\%7812500])\%mod$$
然后套用中国剩余定理就可以了
$ps:$由于同余式有$a\equiv b\mathrm{m}\mathrm{o}\mathrm{d}p\to a\equiv b\mathrm{m}\mathrm{o}\mathrm{d}d$其中$d|p$，所以我们在做快速幂和求幂次前缀和还有斐波那契数列的时候都可以模上$mod=1000000000$，又有模数都一样了之后前缀和与斐波那契数列只用开一个数组就可以得到答案了

#include<bits/stdc++.h>
using namespace std;
const int l1=768;
const int l2=7812500;
const int MOD=1e9;
long long m[]= {512,1953125};
long long f[l2+5];
long long sum[l2+5];
long long quickmod(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*a%MOD;
        b>>=1;
        a=a*a%MOD;
    }
    return ans;
}
void exgcd(long long  a1,long long b,long long &x,long long &y)
{
    if(b==0)
    {
        x=1;
        y=0;
        return ;
    }
    exgcd(b,a1%b,x,y);
    long long t=x;
    x=y;
    y=t-(a1/b)*y;
}
long long CRT(long long a[],long long m[],int n)
{
    long long M=1,ans=0,t,x,y;
    for(int i=0; i<n; i++)
    {
        M*=m[i];
    }
    for(int i=0; i<n; i++)
    {
        t=M/m[i];
        exgcd(t,m[i],x,y);
        ans=(ans+a[i]*x*t)%M;
    }
    return (ans+M)%M;
}
int main()
{
    int n,_m;
    scanf("%d%d",&n,&_m);
    f[0]=0;
    f[1]=sum[1]=1;
    for(int i=2; i<l2+5; i++)
    {
        f[i]=(f[i-1]+f[i-2])%MOD;
        sum[i]=(sum[i-1]+quickmod(f[i],_m))%MOD;
    }
    long long a[2];
    a[0]=((n/l1)*sum[l1]+sum[n%l1])%MOD;
    a[1]=((n/l2)*sum[l2]+sum[n%l2])%MOD;
    long long ans=CRT(a,m,2);
    printf("%lld\n",ans);
    return 0;
}