Final Destination II -- 矩阵快速幂模板题

求f[n]=f[n-1]+f[n-2]+f[n-3]

我们知道

f[n] f[n-1] f[n-2]         f[n-1]  f[n-2]  f[n-3]         1    1     0

0     0       0         =    0          0         0     *      1    0     1

0     0       0               0           0        0            1     0     0

矩阵快速幂就是利用快速幂的思路，去加速转移矩阵的计算，从而给计算带来方便

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#define LL long long
using namespace std;
const LL mod = 1e9+;
struct M
{
    LL a[][];
    void init()
    {
        memset(a, , sizeof(a));
        a[][]=;
        a[][]=;
        a[][]=;
    }
    void init0(){
    memset(a, , sizeof(a));
    }
};
M mul (M a,M b)
{
    M ans;
    for (int i=; i<; i++)
    {
        for (int j=; j<; j++)
        {
            ans.a[i][j]=;
            for (int k=; k<; k++)
            {
                ans.a[i][j]+=a.a[i][k]*b.a[k][j];
                ans.a[i][j]%=mod;
            }
        }
    }
    return ans;
}
M qpow(M a,LL n)
{
    M ans;
    ans.init();
    while(n)
    {
        if (n&)ans=mul(ans,a);
        a=mul(a,a);
        n/=;
    }
    return ans;
}
void output(M a)
{
    for(int i=; i<; ++i)
    {
        for(int j=; j<; ++j)
        {
            cout << a.a[i][j] << " ";
        }
        cout << endl;
    }
}
int main()
{
    int n;
    int t;
    int num=;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        num++;
        M a,b;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
        a.a[][]=;
         //   output(a);
            M k=qpow(a,n);
//            output(a);
//            cout<<endl;
            printf("Case %d:\n",num);
            printf("%lld\n",k.a[][]);
    }
    return ;
}