uva10870(矩陣快速冪)

題意:f(n) = a1 *f(n - 1) + a2 *f(n - 2) + a3 *f(n - 3) + … + ad* f(n - d),  n > d.求f(n)

思路:矩陣快速冪


#include<cstdio>
#include<cstring>
#define ll long long
struct mat {
	ll g[15][15];
}res, ori;
ll n, d, m;
mat mul(mat a, mat b) {
	mat tmp;
	memset(tmp.g, 0, sizeof(tmp.g));
	for(int i = 0; i < d; i++) {
		for(int j = 0; j < d; j++) {
			for(int k = 0; k < d; k++) {
				tmp.g[i][j] = (tmp.g[i][j] + a.g[i][k] * b.g[k][j]) % m;
			}
		}
	}
	return tmp;
}
void cul(ll k){
	while(k) {
		if(k & 1)
			res = mul(res, ori);
		k >>= 1;
		ori = mul(ori, ori);
	}
	printf("%lld\n",res.g[0][0]);
}
int main() {
	while(scanf("%lld%lld%lld",&d,&n,&m) && d) {
		memset(ori.g, 0, sizeof(ori.g));
		memset(res.g, 0, sizeof(res.g));
		for(int i = 0; i < d; i++) {
			scanf("%lld",&ori.g[i][0]);
			ori.g[i][0] %= m;
		}
		for(int i = 0; i < d; i++) {
			scanf("%lld",&res.g[0][d - 1- i]);
			res.g[0][d - 1 - i] %= m;
		}
		if(n <= d) {
			printf("%lld\n",res.g[0][d-n]);
			continue;
		}
		for(int i = 0; i < d - 1; i++) {
			ori.g[i][i + 1] = 1;
		}
		cul(n - d);
	}
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章