洛谷 P1939 【模板】矩阵加速（数列）

 感觉矩阵乘法真的牛逼
关于怎么构造矩阵，可以看这篇博客，自己先不看答案构造一遍
看自己哪里不一样，做一遍就差不多会构造了

矩阵构造

这种矩阵快速可以优化一些比较简单的递推式
构造出一个中间矩阵，然后先做中间矩阵的快速幂
然后乘上初始矩阵就好了

#include<cstdio>
#include<cstring>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
#define _for(i, a, b) for(int i = (a); i <= (b); i++)
using namespace std;

typedef long long ll;
const int MOD = 1e9 + 7;
struct mat
{
	ll m[5][5];
	mat() { memset(m, 0, sizeof(m)); }
};

mat operator * (const mat& a, const mat& b)
{
	mat res;
	_for(i, 1, 3)
		_for(j, 1, 3)
			_for(k, 1, 3)
				res.m[i][j] = (res.m[i][j] + a.m[i][k] * b.m[k][j]) % MOD;
	return res;
}

mat pow(mat a, int b)
{
	mat res;
	_for(i, 1, 3) res.m[i][i] = 1;
	for(; b; b >>= 1)
	{
		if(b & 1) res = res * a;
		a = a * a;
	}
	return res;
}

int main()
{
	int T, n;
	scanf("%d", &T);
	while(T--)
	{
		scanf("%d", &n);
		if(n <= 3) { puts("1"); continue; }
		mat A;
		A.m[1][1] = A.m[1][2] = A.m[3][1] = A.m[2][3] = 1;
		A = pow(A, n - 3);
		printf("%lld\n", (A.m[1][1] + A.m[2][1] + A.m[3][1]) % MOD);
	}
	return 0;
}