菜单 学习猿地 - LMONKEY

VIP

开通学习猿地VIP

尊享10项VIP特权 持续新增

知识通关挑战

打卡带练!告别无效练习

接私单赚外块

VIP优先接,累计金额超百万

学习猿地私房课免费学

大厂实战课仅对VIP开放

你的一对一导师

每月可免费咨询大牛30次

领取更多软件工程师实用特权

入驻
41
0

Blocks

原创
05/13 14:22
阅读数 42398

Blocks

利用递推的想法得到一个递推公式,在本质上它是一个 dp 的题目,但是由于 n 过大,不能使用 dp 解决,所以将递推公式化为矩阵乘法,所以可以用矩阵快速幂进行优化。

// Created by CAD on 2020/2/18.
#include <iostream>
#include <vector>
#define ll long long
using namespace std;

typedef vector<ll> vec;
typedef vector<vec> mat;
const int mod=1e4+7;

mat operator *(mat &a,mat &b){
    mat ans(a.size(),vec(b[0].size()));
    for(int i=0;i<a.size();++i)
        for(int j=0;j<b[0].size();++j)
            for(int k=0;k<b.size();++k)
                ans[i][j]=(ans[i][j]+a[i][k]*b[k][j])%mod;
    return ans;
}
mat qpow(mat x,ll n){
    mat ans(x.size(),vec(x.size()));
    for(int i=0;i<x.size();++i)
        ans[i][i]=1;
    while(n){
        if(n&1) ans=ans*x;
        n>>=1,x=x*x;
    }
    return ans;
}

int main(){
    mat x(3,vec(3));
    x[0][0]=2,x[0][1]=1,x[0][2]=0;
    x[1][0]=2,x[1][1]=2,x[1][2]=2;
    x[2][0]=0,x[2][1]=1,x[2][2]=2;
    int t;cin>>t;
    while(t--){
        int n;cin>>n;
        mat ans=qpow(x,n);
        cout<<ans[0][0]<<'\n';
    }
}

发表评论

0/200
41 点赞
0 评论
收藏