【矩阵快速幂模板】

【矩阵快速幂 O(log(n))】

【NYOJ301】580ms

递推求值

时间限制: 1000 ms  |            内存限制: 65535 KB
难度: 4
描述

给你一个递推公式:

f(x)=a*f(x-2)+b*f(x-1)+c

并给你f(1),f(2)的值,请求出f(n)的值,由于f(n)的值可能过大,求出f(n)对1000007取模后的值。

注意:-1对3取模后等于2

输入
第一行是一个整数T,表示测试数据的组数(T<=10000)
随后每行有六个整数,分别表示f(1),f(2),a,b,c,n的值。
其中0<=f(1),f(2)<100,-100<=a,b,c<=100,1<=n<=100000000 (10^9)
输出
输出f(n)对1000007取模后的值
样例输入
2
1 1 1 1 0 5
1 1 -1 -10 -100 3
样例输出
5
999896

 

/*
构造矩阵
X
{0 a 0
 1 b 0
 0 1 1}
使得{f(x-2),f(x-1),c} * X = {f(x-1),a*f(x-2)+b*f(x-1)+c, c} == {f(x-1),f(x),c}
{f(1),f(2),c}乘上N-1次X后就可得出{f(n),f(n+1),c}  f(n)就是最终答案
也就{f(1),f(2),c} * X^n-1 = {f(n),f(n+1),c};
*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
#define rep(i,a,n) for(int i = a; i < n; i++)
#define repe(i,a,n) for(int i = a; i <= n; i++)
#define per(i,n,a) for(int i = n; i >= a; i--)
#define clc(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
typedef long long LL;
#define MOD 1000007
struct MATRIX{//矩阵
	LL num[3][3];
	MATRIX(LL a1,LL a2,LL a3, LL b1, LL b2, LL b3, LL c1, LL c2, LL c3)
	{
		num[0][0] = a1, num[0][1] = a2, num[0][2] = a3;
		num[1][0] = b1, num[1][1] = b2, num[1][2] = b3;
		num[2][0] = c1, num[2][1] = c2, num[2][2] = c3;
	}
};

MATRIX mul(const MATRIX& a, const MATRIX& b)//矩阵a*b
{
	MATRIX ans=MATRIX(0,0,0,0,0,0,0,0,0);
	rep(i,0,3)
	{
		rep(j,0,3)
		{
			rep(k,0,3)
				ans.num[i][j] = (ans.num[i][j]+(a.num[i][k]*b.num[k][j]%MOD+MOD)%MOD)%MOD;
		}
	}
	return ans;
}
MATRIX pow_mod(MATRIX x, int n)//快速幂
{
	MATRIX ans = MATRIX(1,0,0,0,1,0,0,0,1);//任何3*3矩阵乘以这个单位矩阵不会改变值
	while(n)
	{
		if(n&1) ans = mul(ans,x);
		x = mul(x,x);
		n >>= 1;
	}
	return ans;
}
int main()
{
#ifdef SHY
	freopen("e:\\1.txt", "r", stdin);
#endif
	int t;
	scanf("%d%*c", &t);
	while(t--)
	{
		int f1,f2,a,b,c,n;
		scanf("%d %d %d %d %d %d%*c", &f1, &f2, &a, &b, &c, &n);
		MATRIX x = MATRIX(0,a,0,1,b,0,0,1,1);//构造出的矩阵
		x = pow_mod(x,n-1);
		int ans = ((f1*x.num[0][0]%MOD+MOD)%MOD+(f2*x.num[1][0]%MOD+MOD)%MOD+(c*x.num[2][0]%MOD+MOD)%MOD)%MOD;
		//只要求出[0][0]即可
		printf("%d\n", ans);
	}
	return 0;
}


 


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值