BUPT Summer Journey #test5 D

 

D. Fibnacci 2014新生暑假个人排位赛05

时间限制 1000 ms   内存限制 65536 KB

题目描述

已知函数f,f(0)=f(1)=1,对于所有n>=2，f(n)=f(n-1)+f(n-2)+n。给定n，求f(n)。

输入格式

第一行一个整数t,表示组数,t<1000。之后t行每行一个整数n,0<=n<=1e9。

输出格式

对于每组测试组，输出一行f(n)%1000000007.

输入样例

6
1
2
3
4
5
1000000000

输出样例

1
4
8
16
29
999999980

 

思路：

首先。我们将其转化为线性递推关系。f(n)=f(n-1)+f(n-2)+n,f(n-1)=f(n-2)+f(n-3)+(n-1).得f(n)=2*f(n-1)-f(n-3)。然后用矩阵快速幂即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#define mod 1000000007
using namespace std;
long long n,k,b[12][12],t[12][12],z[12][12],c[100],f[100];
void ff(long long x[][12],long long y[][12])
{
    memset(z,0,sizeof(z));
    for(int i=1;i<=n+1;i++)
        for(int j=1;j<=n+1;j++)
            for(int p=1;p<=n+1;p++)z[i][j]=(z[i][j]+x[i][p]*y[p][j])%mod;
    memcpy(x,z,sizeof(z));
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(b,0,sizeof(b));
        n=3;b[1][n]=-1;;b[3][n]=2;b[4][n]=1;f[1]=1;f[2]=4;f[3]=8;
        for(int i=1;i<=n;i++)b[i+1][i]=1;
        b[n+1][n+1]=1;
        memcpy(t,b,sizeof(b));
        //for(int i=1;i<=n+1;i++){for(int j=1;j<=n+1;j++)printf("%d ",b[i][j]);printf("\n");}
        scanf("%d",&k);
        if(k==1){printf("%lld\n",f[1]);continue;}
        if(k==2){printf("%lld\n",f[2]);continue;}
        if(k==3){printf("%lld\n",f[3]);continue;}
        int i=k-3;
        memset(c,0,sizeof(c));
        while(i>0){c[++c[0]]=i%2;i/=2;}
        for(int i=c[0]-1;i>=1;i--)
        {
            ff(b,b);
            if(c[i]==1)ff(b,t);
        }
        long long s=b[n+1][n];
        for(int i=1;i<=n;i++)s=(s+f[i]*b[i][n])%mod;
        s=(s+mod)%mod;
        printf("%lld\n",s);
    }
    return 0;
}