构建矩阵及矩阵快速幂

(Problem - 3306 (hdu.edu.cn))

 思路：构造矩阵，然后矩阵快速幂

大致过程如下：（从课件里截过来的，其实很简单）

 

 ac代码：

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
const int MAX = 4;
const int MOD = 10007;
typedef struct {
	int m[MAX][MAX];
}Mit;//创建出相应大小的矩阵
Mit I = { 1,0,0,0,
		0,1,0,0,
		0,0,1,0,
		0,0,0,1 };//I一直是一个MAX*MAX的单位阵
int n, x, y;
Mit Mul(Mit a, Mit b)//快速乘带矩阵进去
{
	Mit c;
	for (int i = 0; i < MAX; i++) {
		for (int j = 0; j < MAX; j++) {
			c.m[i][j] = 0;
			for (int k = 0; k < MAX; k++) {
				c.m[i][j] += (a.m[i][k] * b.m[k][j]) % MOD;
			}
			c.m[i][j] %= MOD;
		}
	}
	return c;
}
Mit quick(Mit p, Mit I, int n)//快速幂带矩阵进去
{
	Mit m = p, b = I;
	while (n >= 1) {
		if (n & 1) b = Mul(b, m);
		n = n >> 1;
		m = Mul(m, m);
	}
	return b;
}
int main()
{
	while (cin >> n >> x >> y) {
		x %= MOD;
		y %= MOD;
		Mit p = { 1,(x * x) % MOD,(y * y) % MOD,(2 * x * y) % MOD,0,(x * x) % MOD,(y * y) % MOD,(2 * x * y) % MOD,0,1,0,0,0,x,0,y };//p就是构造出来的矩阵，记得多取模要不然会爆
		Mit S;
		S = quick(p, I, n - 1);
		long long ans;
		ans = (S.m[0][0] * 2) % MOD;
		ans += S.m[0][1] % MOD;
		ans += S.m[0][2] % MOD;
		ans += S.m[0][3] % MOD;
		ans %= MOD;
		cout << ans << endl;
	}
	return 0;
}