矩阵加速(数列)
矩阵优化是个很不错的工具,但struct的坑太多了!好难改.
首先我们知道初始状态
[
f
[
3
]
f
[
2
]
f
[
1
]
]
\begin{bmatrix}f[3]&\\f[2]&\\f[1]\end{bmatrix}
⎣⎡f[3]f[2]f[1]⎦⎤
[
f
[
i
]
f
[
i
−
1
]
f
[
i
−
2
]
]
\begin{bmatrix}f[i]&\\f[i-1]&\\f[i-2]\end{bmatrix}
⎣⎡f[i]f[i−1]f[i−2]⎦⎤ 推到
[
f
[
i
+
1
]
f
[
i
]
f
[
i
−
1
]
]
\begin{bmatrix} f[i+1]&\\f[i]&\\f[i-1] \end{bmatrix}
⎣⎡f[i+1]f[i]f[i−1]⎦⎤
则可以构造矩阵
[
1
0
1
1
0
0
0
1
0
]
\begin{bmatrix}1&0&1&\\1&0&0&\\0&1&0\end{bmatrix}
⎣⎡110001100⎦⎤
初始矩阵
我们发现,有一种特殊的矩阵,就像1一样,任何矩阵乘它都等于本身,所以我们可以用这样的矩阵作初始矩阵
[
1
⋯
0
⋮
⋱
⋮
0
⋯
1
]
\begin{bmatrix}1&\cdots&0\\\vdots&\ddots&\vdots\\0&\cdots&1\end{bmatrix}
⎣⎢⎡1⋮0⋯⋱⋯0⋮1⎦⎥⎤
即
for(int i=0;i<size;i++) matrix[i][i]=1;
这样的矩阵
最后算答案的时候就是 [ f [ 3 ] f [ 2 ] f [ 1 ] ] ∗ 得 到 的 a n s 矩 阵 就 行 了 \begin{bmatrix}f[3]&\\f[2]&\\f[1]\end{bmatrix}*得到的ans矩阵就行了 ⎣⎡f[3]f[2]f[1]⎦⎤∗得到的ans矩阵就行了,但对于本题来说 , [ f [ 3 ] f [ 2 ] f [ 1 ] ] 都 是 1 , 所 以 直 接 相 加 就 可 以 了 \begin{bmatrix}f[3]&\\f[2]&\\f[1]\end{bmatrix}都是1,所以直接相加就可以了 ⎣⎡f[3]f[2]f[1]⎦⎤都是1,所以直接相加就可以了
P.S :struct 有点坑,每次定义一个struct都需要初始化,还有就是struct不要用1ll乘,直接long long 就行了,至于为什么,我也不知道
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MOD=1e9+7;
int n;
struct Matric{
ll a[3][3];//不开long long会炸
}res,ans;
void init(){//类型一定是void,因为初始化不需要返回数组名
memset(res.a,0,sizeof(res.a));
memset(ans.a,0,sizeof(ans.a));
res.a[0][0]=res.a[0][2]=res.a[1][0]=res.a[2][1]=1;
ans.a[0][0]=ans.a[1][1]=ans.a[2][2]=1;
}
Matric mul(Matric x,Matric y){
Matric anss;
memset(anss.a,0,sizeof(anss.a));/*结构体一定要初始化,不然初始值会是一些很神奇的数据*/
for(int i=0;i<3;i++)
for(int j=0;j<3;j++)
for(int k=0;k<3;k++){
anss.a[i][k]=(anss.a[i][k]+(x.a[i][j]%MOD)*(y.a[j][k]%MOD)%MOD)%MOD;
}
return anss;
}
void ksm(ll pow){//快速幂也不需要数组名
while(pow){
if(pow&1) ans=mul(ans,res);
res=mul(res,res);
pow>>=1;
}
}
int main(){
int T;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
if(n<4){
puts("1");
continue;
}
init();
ksm(n-3);
printf("%d\n",(ans.a[0][0]+ans.a[0][1]+ans.a[0][2])%MOD);
}
}