矩阵快速幂

洛谷 P1962 斐波那契数列
洛谷 P1939 矩阵加速
洛谷 P1397 矩阵游戏
因为个人能力问题,在这里给出别人的P1397题解链接
根据题目:
a[1]=a[2]=a[3]=1
a[x]=a[x-3]+a[x-1] (x>3)
我们可以用下面的矩阵a和矩阵b相乘得到c
a:
1 1 1
*
b:
0 0 1
1 0 0
0 1 1
c:
2 3 4
然后我们就可以矩乘了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define ll long long
#define mod 1000000007
using namespace std;
int n,tt;
struct matrix{
    ll m[4][4];
}a,b,t;
matrix mi(matrix a, matrix b){
    matrix t;//矩阵乘法
    for(int i=1;i<=3;++i)
        for(int j=1;j<=3;++j){
            t.m[i][j]=0;
            for(int k=1;k<=3;++k)
                t.m[i][j]=(t.m[i][j]+a.m[i][k]*b.m[k][j])%mod;
        }
    return t;
}
matrix fast_m(matrix a,ll b){//矩阵快速幂
    matrix s=a,w=a; b--;
    while(b) {if(b&1) s=mi(w,s);w=mi(w,w);b>>=1;}
    return s;
}
int main(){
    scanf("%d",&n);
    b.m[1][1]=1;b.m[1][2]=1;b.m[1][3]=1;
    for(int i=1;i<=n;++i){
        scanf("%d",&tt);
        a.m[1][1]=0;a.m[1][2]=0;a.m[1][3]=1;
            a.m[2][1]=1;a.m[2][2]=0;a.m[2][3]=0;
            a.m[3][1]=0;a.m[3][2]=1;a.m[3][3]=1;//以上三行是赋初始值
        if(tt==1){ cout<<"1\n";continue;}
        if(tt==2){ cout<<"1\n";continue;}
        if(tt==3){ cout<<"1\n";continue;}//tt等于1,2,3的情况直接输出
        tt-=3;//从第4个开始,减去3;
        a=fast_m(a,tt);//得到矩乘后的一个新的矩阵
        for(int i=1;i<=1;++i)//再将这个矩阵与上述所说的a矩阵相乘,就会得到最后的c矩阵,在代码中a矩阵为b矩阵
            for(int j=1;j<=3;++j){
                t.m[i][j]=0;
                for(int k=1;k<=3;++k)
                    t.m[i][j]=(t.m[i][j]+b.m[i][k]*a.m[k][j])%mod;
            }
        printf("%lld\n",t.m[1][3]);//因为我们每次矩乘一次只会得到一个新的数字,并且是处于之前所讲的c矩阵的第三位,所以在这里输出最后一位
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值