P1939 【模板】矩阵加速(数列)

传送门
根据题目的意思,还有瞎jb凑,凑出来这个矩阵
|f[n],f[n-1],f[n-2]|=|f[n-1],f[n-2],f[n-3]|*|1,0,1,|
                  |1,0,0|
                  |0,1,0|
还是老套路矩阵快速幂就可以AC辣

#include <cstdio>
#include <iostream> 
#include <cstring>
#define ll long long 
using namespace std;
const int p=1e9+7;
ll ans[2][4];
ll x[4][4];
ll dx[4][4];
void init()
{
    memset(ans,0,sizeof(ans));
    memset(x,0,sizeof(x));
    ans[1][1]=1,ans[1][2]=1,ans[1][3]=1;
    x[1][1]=1,x[1][3]=1,x[2][1]=1,x[3][2]=1;
}
void ans_cf()
{
    for(int i=1;i<=3;i++)
     dx[1][i]=ans[1][i],ans[1][i]=0;
    for(int i=1;i<=1;i++)
     for(int j=1;j<=3;j++)
      for(int k=1;k<=3;k++)
       ans[i][j]=(ans[i][j]+(dx[i][k]*x[k][j])%p)%p;
}
void x_cf()
{
    for(int i=1;i<=3;i++)
     for(int j=1;j<=3;j++)
      dx[i][j]=x[i][j],x[i][j]=0;
    for(int i=1;i<=3;i++)
     for(int j=1;j<=3;j++)
      for(int k=1;k<=3;k++)
       x[i][j]=(x[i][j]+(dx[i][k]*dx[k][j])%p)%p;
}
void fast_pow(int k)
{
    while(k)
    {
        if(k%2) ans_cf();
        k/=2;
        x_cf();
    }
}
int main()
{
    ll n;
    scanf("%lld",&n);
    while(n--)
    {
        init();
        ll w;
        scanf("%lld",&w);
        if(w<=3)
        {
            printf("1\n");
            continue;
        }
        fast_pow(w-3);
        printf("%lld\n",ans[1][1]%p);

    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值