洛谷P1939 矩阵快速幂模板

关注

原创最新推荐文章于 2026-07-11 09:25:00 发布 724 阅读 AI 写同款· GEO 优化 ›

洛谷 P1939 【模板】矩阵加速（数列）：优化递推式的方法——矩阵快速幂 在大多数情况下，O(n)的效率都是值得骄傲的，然而，有时候并不是，比如如何在一秒钟内算出一个递推式的第1e9项，很明显O(n)不行了。然而常数级又不太现实，除非你的数学非常好，这题又比较简单，你推了一个特征方程的通项公式…… 所以考虑log的做法：矩阵快速幂 如果你还不知道矩阵快速幂是什么，请走这边：传送门对于这道题，嗯，模板嘛，已经告诉你了式子，就只需要考虑矩阵了，对于整个过程，我们阅读详情

矩阵快速幂解线性递推式：

$a_{n}=a_{n-1}+a_{n-3}$

Solution：

一般构造矩阵的时候都构造一个连续的矩阵，且这个矩阵元素下标都 $+ 1$ 后两个矩阵恰好覆盖了递推式所涉及的所有项，设递推式跨越的下标长度为 $l e n$ ，这个长度一般是 $l e n - 1$ ，譬如当前构造矩阵
$a_{n},a_{n-1},a_{n-2}]$
我们需要构造出一个矩阵，使得这个上式矩阵乘构造矩阵后得到
$a_{n+1},a_{n},a_{n-1}]$

矩阵乘法：

对于两个矩阵 $a, b$ 相乘，结果矩阵的 $i$ 行 $j$ 列为
$\sum_{k=1}^{n}a_{i,k}b_{k,j}$
即 $a$ 的 $i$ 行与 $b$ 的 $j$ 列的所有元素对应相乘相加

单位矩阵：

是主对角线元素为 $1$ ，其余元素为 $0$ 的矩阵，即满足下式的矩阵
$a_{i,j}=\begin{cases} 0 & i\neq j\\ 1 & i=j \end{cases}$

根据矩阵乘法的规则构造矩阵为
$\left[ \begin{matrix} 1 & 1 & 0\\ 0 & 0 & 1\\ 1 & 0 & 0 \end{matrix} \right]$
对此矩阵快速幂即可

需要注意的是，单位矩阵是主对角线元素为 $1$ 的矩阵，其他元素为 $0$ ；而不是都是 $1$ 的矩阵

// #include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#include<bitset>
#include<map>
using namespace std;

using ll=long long;
const int N=1e6+5,inf=0x3fffffff;
const long long INF=0x3fffffffffffffff,mod=1e9+7;

struct matrix
{
    ll a[3][3];
    void initial(){memset(a,0,sizeof(a));}
    void build()
    {
        initial();
        for(int i=0;i<3;i++) a[i][i]=1;
    }
    ll* operator[](int i) const {
        return (ll*)&a[i][0];
    }
    matrix operator*(const matrix& x) const
    {
        matrix ret; ret.initial();
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
                for(int k=0;k<3;k++) ret[i][j]=(ret[i][j]+a[i][k]*x[k][j]%mod)%mod;
        return ret;
    }
};

matrix qpow(matrix a,ll b)
{
    matrix ret,base=a; ret.build();
    while(b)
    {
        if(b&1) ret=ret*base;
        base=base*base;
        b>>=1;
    }
    return ret;
}

int main()
{
    matrix tmp={{
        {1,1,0},
        {0,0,1},
        {1,0,0}
    }},ans={{
        {1,1,1},
        {0,0,0},
        {0,0,0}
    }};
    int t; cin>>t;
    while(t--)
    {
        ll n; cin>>n;
        if(n<=3) cout<<1<<'\n';
        else cout<<(ans*qpow(tmp,n-3)).a[0][0]<<'\n';
    }
    return 0;  
}