矩阵快速幂 P1939 【模板】矩阵加速（数列）

题目描述

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。

输出格式：

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

输入输出样例

输入样例#1： 复制

3
6
8
10

输出样例#1： 复制

4
9
19

说明

对于30%的数据 n<=100；

对于60%的数据 n<=2*10^7；

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

由于 f[n]=f[n-1]+f[n-3] 构造矩阵时要满足这个式子， 所以有{  f[n] ，f[n-1]， f[n-2] }={ f[n-1]，f[n-2]，f[n-3] } X 矩阵A      （不是差乘）

f[n]=1*f[n-1]+0*f[n-2]+1*f[n-3] 所以第1列为   1  0  1           是列

同理由于已经满足了 f[n]的关系式所以不去满足剩下的f[n-1] f[n-2] 有

f[n-1] = 1*f[n-1] + 0*f[n-2]+  0*f[n-3]     第2列为1 0 0

f[n-2]=0*f[n-1] + 1*f[n-2] + 0*f[n-3]       第3列为0 1 0

A =   1     1     0

         0     0     1

         1     0     0

#include <math.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <cstdio>
#include <list>
#include <deque>
#include <set>
#include <vector>
#include <map>
#include <cstring>
#include <iomanip>
#define max(a,b) (a>b?a:b)
#define LL long long
using namespace std;
const int mod=1e9+7;
struct node
{
    LL a[120][120];
};
int n;
node getnum(node x,node y)
{
    node z;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        z.a[i][j]=0;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            for(int k=0;k<n;k++)
            {
                z.a[i][j]=(z.a[i][j]%mod+x.a[i][k]*y.a[k][j]%mod)%mod;
            }
            return z;
}

node  Pow(node a,LL b)
{
   /* for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        {
        z.a[i][j]=0;
        }*/
        node ans;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
            {
                if(i==j)ans.a[i][j]=1;
                else ans.a[i][j]=0;
            }

    while(b!=0)
    {
        if(b&1!=0)
          ans=getnum(ans,a);
          a=getnum(a,a);
           // getnum(a,a,z);
        //getnum(a,a,a);
        b>>=1;
    }
return ans;
}
//int num[1000];
node num;
int main()
{
    ios::sync_with_stdio(false);
   LL n1;
    int t;
    cin>>t;
    while(t--)
    {

    cin>>n1;
    int a=1,b=1,c=1;
    n=3;
   for(int i=0;i<3;i++)
    for( int j=0;j<3;j++)
    {
        num.a[i][j]=0;
        if((i==0&&j==0)||(i==0&&j==1)||(i==1&&j==2)||(i==2&&j==0))
        num.a[i][j]=1;
    }
    node ans=Pow(num,n1);
   /* for(int i=0;i<3;i++)
        for(int j=0;j<3;j++)
        {
            if(j==2)cout<<ans.a[i][j]<<endl;
            else cout<<ans.a[i][j]<<" ";
        }*/

        cout<<ans.a[0][1]<<endl;

    }
  return 0;
}

结果为什么是a[0][1] 由于算的是n+1的  当然输出矩阵也可以发现