【luogu 1939】 线性递推 + 矩阵快速幂

题目:a_n = a_{n-1} + a_{n-3},\ a_1 = a_2 = a_3 = 1

不知道为什么今天下午做矩阵快速幂上头。

由题意:

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

我们只需要找到 \left[ \begin{matrix} a_{n-1} & a_{n-2} & a_{n-3} \end{matrix} \right ] 与 \left[ \begin{matrix} a_n&a_{n-1} &a_{n-2} \end{matrix} \right ]的线性递推关系即可。

我们可以得到如下三个线性方程

a_n = 1 * a_{n-1} + 0 * a_{n-2} + 1 * a_{n-3}

a_{n-1} = 1 * a_{n-1} + 0 * a_{n-2} + 0 * a_{n-3}

a_{n-2} = 0 * a_{n-1} + 1 * a_{n-2} + 0 * a_{n-3}

于是

\[ \begin{matrix} a_{n-1}&a_{n-2}&a_{n-3} \end{matrix} \] \left\[ \begin{matrix} 1 & 1 & 0\\ 0 & 0 & 1\\ 1 & 0 & 0\\ \end{matrix} \right] = \left\[\ \begin{matrix} a_{n} & a_{n-1} & a_{n-2} \end{matrix} \right]

最后我们就可以使用矩阵快速幂求解了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;
struct matrix {
	ll a[4][4];
	matrix() {
		memset(a, 0, sizeof(a));
	}
	matrix operator * (const matrix& b) {
		matrix res;
		for(int i = 1; i <= 3; i++) {
			for(int j = 1; j <= 3; j++) {
				for(int k = 1; k <= 3; k++) {
					res.a[i][j] = (res.a[i][j] % mod + a[i][k] % mod * b.a[k][j] % mod + mod) % mod;
				}
			}
		}
		return res;
	}
};
ll power(ll n)
{
	matrix ans, base;
	for(int i = 1; i <= 3; i++) {
		for(int j = 1; j <= 3; j++) {
			ans. a[i][j] = base.a[i][j] = 0;
		}
	}
	ans.a[1][1] = ans.a[1][2] = ans.a[1][3] = 1;
	base.a[1][1] = base.a[1][2] = 1;
	base.a[2][3] = 1;
	base.a[3][1] = 1;
	while(n) {
		if(n & 1) ans = ans * base;
		base = base * base;
		n >>= 1;
	}
	return ans.a[1][1];
}
int main()
{
	int t;
	ll n;
	scanf("%d", &t);
	while(t--) {
		scanf("%lld", &n);
		if(n == 1 || n == 2 || n == 3) printf("1\n");
		else printf("%lld\n", power(n-3));
	}

}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值