NKOJ-3712 数列<L特供版>

P3712数列
时间限制 : - MS 空间限制 : 65536 KB
评测说明 : 时限1000ms
问题描述

a[1]=a[2]=a[3]=1
a[x]=a[x-3]+a[x-1]  (x>3)
求a数列的第n项对1000000007（10^9+7）取余的值。

输入格式

第一行一个整数T，表示询问个数。
以下T行，每行一个正整数n。

输出格式

每行输出一个非负整数表示答案。

样例输入

3
6
8
10

样例输出

4
9
19

提示

对于30%的数据 n<=100；
对于60%的数据 n<=2*10^7；
对于100%的数据 T<=100，n<=2*10^9

为什么感觉最近都没有可吐槽的内容了

题解

这道题目好像很简单…纯粹是瞎写矩阵乘法导致的WA

首先，公式题目就给了，所以连推导都不需要
另外，由于这里涉及到三项，所以我们就需要一个3*3的矩阵
第一个矩阵定义为

a[n]  a[n-1]  a[n-2]

0      0        0

0      0        0

由于a[n+1]=a[n]+a[n-2]
所以第二个矩阵第一列为

1 （a[n]的系数)
0 （a[n-1]的系数)
1 （a[n-2]的系数)

由于变换后,第一个矩阵第一行为

a[n+1] a[n] a[n-1]

所以完整的第二个矩阵为

1 1 0
0 0 1
1 0 0

我们也可以这么理解

a[n+1]= 1 *a[n]+ 0 *a[n-1]+ 1 *a[n-2]
a[n]  = 1 *a[n]+ 0 *a[n-1]+ 0 *a[n-2]
a[n-1]= 0 *a[n]+ 1 *a[n-1]+ 0 *a[n-2]

仔细观察每一行式子的系数，单独分离出来就是

1 0 1
1 0 0 
0 1 0
  |
把第一行竖着看成第一列 例如 1 0 1->1
第二行看成第二列                 0
第三行看成第三列                 1
  |
  V
1 1 0
0 0 1
1 0 0 

然后你会发现，这些系数旋转九十度之后就是第二个矩阵了
仔细思考一下为什么吧（结合刚才提到的系数）

矩阵都出来了，剩余的应该就不需要多说了吧

附上代码

#include <iostream>
#include <cstdio>
#include <cstring>
#define mod 1000000007
using namespace std;
typedef long long mat[3][3];

mat ori={{1,1,1},{0,0,0},{0,0,0}},add={{1,1,0},{0,0,1},{1,0,0}};

struct trix
{
    mat x;
    trix(){memset(x,0,sizeof(x));}
    trix operator *(const trix & y)
    {
        trix ans=trix();
        for(int a=0;a<3;a++)
        for(int b=0;b<3;b++)
        for(int m=0;m<3;m++)
            ans.x[a][b]+=(x[a][m]*y.x[m][b])%mod;
        for(int a=0;a<3;a++)
        for(int b=0;b<3;b++)
            ans.x[a][b]=ans.x[a][b]%mod;        
        return ans;
    }
    trix operator =(const mat y){memcpy(x,y,sizeof(x));}
    trix operator =(const trix & y){memcpy(x,y.x,sizeof(x));}

    void out(){printf("%lld\n",x[0][0]%mod);}
};

trix res,use;

void multi(int t)
{
    res=ori;use=add;
    while(t)
    {
        if(t&1)res=res*use;
        use=use*use;
        t>>=1;
    }
    res.out();
}

int main()
{
    int T,ask;
    scanf("%d",&T);
    for(int t=1;t<=T;t++)
    {
        scanf("%d",&ask);
        if(ask<=3)printf("1");
        else multi(ask-3);
    }
}