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

板子传送门
矩阵快速幂学完当然要去搞一搞矩阵加速啦
(矩阵加速相对于矩阵快速幂来说就是多了一个构造矩阵的过程)
关于怎样来构造矩阵,这位大佬讲的很好呢
构造出矩阵之后,我们再去用矩阵快速幂乘出来,取[1,1]就好了呃

//f[i]=f[i-1]+f[i-3]
//f[1]=f[2]=f[3]=1
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#define ll long long

using namespace std;

const ll mod=1e9+7;

struct Matrix{
	ll mx[4][4];
}mat;

int t,n;

inline Matrix Mul(Matrix a,Matrix b){
	Matrix t;
	for(int i=1;i<=3;++i)//这个函数里的所有循环一开始我给写成了n于是乎......WA/TLE/RE到飞起
		for(int j=1;j<=3;++j)
			t.mx[i][j]=0;
	for(int k=1;k<=3;++k)
		for(int i=1;i<=3;++i)
			for(int j=1;j<=3;++j)
				t.mx[i][j]=(t.mx[i][j]+a.mx[i][k]*b.mx[k][j]%mod)%mod;
	return t;
}

inline Matrix quick(Matrix a,int pow){
	if(n<=1) return a;
	Matrix t=a;--pow;
	while(pow){
		if(pow&1) t=Mul(t,a);
		a=Mul(a,a);pow>>=1;
	}
	return t;
}

int main(){
	memset(mat.mx,false,sizeof(mat.mx));
	mat.mx[1][1]=1;mat.mx[3][1]=1;
	mat.mx[1][2]=1;mat.mx[2][3]=1;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		Matrix ans;memset(ans.mx,false,sizeof(ans.mx));
		ans=quick(mat,n-1);
		printf("%lld\n",ans.mx[1][1]%mod);
	}
	return 0;
}
posted @ 2018-09-13 20:49  Phecda  阅读(162)  评论(0编辑  收藏  举报

Contact with me