题目1081：递推数列

原题地址

题目描述：

给定a0,a1,以及an=p*a(n-1) + q*a(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。

输入：

输入包括5个整数：a0、a1、p、q、k。

输出：

第k个数a(k)对10000的模。

样例输入：

20 1 1 14 5

样例输出：

8359

#include <cstdio>
#define Mod 10000

typedef struct{
	int a,b,c,d;
}M;

M o,m;
int a0,a1,k;
//两个普通矩阵相乘
M mul(M m, M n){
	M t;
	t.a = (m.a * n.a + m.b * n.c) % Mod;
	t.b = (m.a * n.b + m.b * n.d) % Mod;
	t.c = (m.c * n.a + m.d * n.c) % Mod;
	t.d = (m.c * n.b + m.d * n.d) % Mod;
	return t;
}

M pow(){
	m.a = m.d = 1;
	m.b = m.c = 0;
	//二分幂乘,求矩阵o的k次方
	while(k){
		if(k&1) m = mul(m,o);
		o = mul(o,o);
		k >>= 1;
	}
	return m;
}


int main(){
	//a0,a1,p,q,k
	while(~scanf("%d%d%d%d%d",&a0,&a1,&o.a,&o.b,&k)){
		o.c = 1;
		o.d = 0;
		if(k==0)
			printf("%d\n", a0);
		else if(k == 1)
			printf("%d\n",a1);
		else{
		k--;
		o = pow();
		printf("%d\n", (o.a * a1 + o.b * a0) % Mod);
		}
	}
	return 0;
}