矩阵加速(数列)

37 篇文章 1 订阅

矩阵加速(数列)
矩阵优化是个很不错的工具,但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[i1]f[i2] 推到 [ 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[i1]
则可以构造矩阵 [ 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} 1001

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);
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值