求fibonacci数列前N个数的K次方和。
通项公式:F[n]=((1+sqrt(5))/sqrt(5)-(1-sqrt(5))/sqrt(5))/sqrt(5)。
有点乱,不过由于可以保证最后的结果是一个整数,所有所有的根号都可以化为整数进行取模和逆元运算。
首先解二次同余方程,X^2=n (mod M),显然,当n=5的时候,X就可以相当于5了。
后面的都可以替换掉。
然后就变成了 F[n]=(A^n+B^n)*C。
这里通过二项式展开,分别对每一项进行计算,同时又可以发现,每一项求和其实是一个等比数列,于是,就可以直接搞了。
召唤代码君:
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #define M 1000000009 #define maxn 100100 typedef long long ll; using namespace std; ll w,a,m,root,inv; ll A[maxn],B[maxn]; ll n,k,ans,cur,now; ll AAA,BBB; int T; struct twice{ ll A,B; twice() {} twice(ll AA,ll BB) { A=AA,B=BB; } void mul(twice T){ ll aa=A*T.A+(B*T.B)%M*w,bb=A*T.B+B*T.A; A=aa%M,B=bb%M; } }; ll power(ll A,ll B,ll C) { ll tot=1; while (B){ if (B&1) tot=tot*A%C; A=A*A%C,B>>=1; } return tot; } twice power(twice T,ll y) { twice C(1,0); while (y){ if (y&1) C.mul(T); T.mul(T),y>>=1; } return C; } ll getroot() { for (;;){ a=rand()%M; w=(a*a-5+M)%M; if (power(w,M/2,M)!=1) break; } return power(twice(a,1),(M+1)/2).A; } void _init() { root=getroot(); A[0]=B[0]=1; for (int i=1; i<maxn; i++){ A[i]=(A[i-1]*i)%M; B[i]=power(A[i],M-2,M); } inv=B[2]; AAA=(1+root)*inv%M; BBB=(M+1-root)*inv%M; } int main() { _init(); scanf("%d",&T); while (T--){ scanf("%lld%lld",&n,&k); ans=0; for (int i=0; i<=k; i++){ cur=A[k]*(B[i]*B[k-i]%M)%M; if (i&1) cur=M-cur; now=power(AAA,k-i,M)*power(BBB,i,M)%M; if (now>1) now=((power(now,n+1,M)-now)*power(now-1,M-2,M)%M+M)%M; else now=now*n%M; (ans+=cur*now)%=M; } ans=ans*power(root,k*(M-2),M)%M; printf("%d\n",(int)ans); } return 0; }