Week14 作业 C - Q老师的考验（必做） HDU - 1757 A Simple Math Problem

Q老师 对数列有一种非同一般的热爱，尤其是优美的斐波那契数列。

这一天，Q老师 为了增强大家对于斐波那契数列的理解，决定在斐波那契的基础上创建一个新的数列 f(x) 来考一考大家。数列 f(x) 定义如下：

当 x < 10 时，f(x) = x；
当 x ≥ 10 时，f(x) = a0 * f(x-1) + a1 * f(x-2) + a2 * f(x-3) + …… + a9 * f(x-10)，ai 只能为 0 或 1。

Q老师 将给定 a0～a9，以及两个正整数 k m，询问 f(k) % m 的数值大小。

聪明的你能通过 Q老师 的考验吗？

Input
输出文件包含多组测试用例，每组测试用例格式如下：

第一行给定两个正整数 k m。（k < 2e9, m < 1e5）

第二行给定十个整数，分别表示 a0～a9。

Output
对于每一组测试用例输出一行，表示 f(k) % m 的数值大小。

Sample Input
10 9999
1 1 1 1 1 1 1 1 1 1
20 500
1 0 1 0 1 0 1 0 1 0
Sample Output
45
104

#include<iostream>
#include<string.h>
using namespace std;

int m;
const int N = 10;
struct Matrix {
	int x[N][N];
	Matrix operator*(const Matrix& t) const {
		Matrix rs;
		for (int i = 0; i < N; ++i) {
			for (int j = 0; j < N; ++j) {
				rs.x[i][j] = 0;
				for (int k = 0; k < N; ++k) {
					rs.x[i][j] += x[i][k] * t.x[k][j];
					rs.x[i][j] = rs.x[i][j] % m;
				}
			}
		}
		return rs;
	}
	Matrix() {
		memset(x, 0, sizeof(x));
	}
	Matrix(const Matrix& t) {
		memcpy(x, t.x, sizeof(x));
	}
};

Matrix qpow(Matrix m, int n) {
	Matrix ret;
	for (int i = 0; i < 10; i++) {
		ret.x[i][i] = 1;
	}
	while (n) {
		if (n & 1) {
			ret = m * ret;
		}
		m = m * m;
		n >>= 1;
	}
	return ret;

}

int main() {
	int k;
	while (scanf("%d%d", &k, &m) != EOF) {
		int L[10];
		for (int i = 0; i < 10; i++) {
			cin >> L[i];
		}
		Matrix m1;
		for (int i = 0; i < 10; i++) {
			m1.x[0][i] = L[i];
		}
		for (int i = 1; i < 10; i++) {
			m1.x[i][i - 1] = 1;
		}
		Matrix mm = qpow(m1, k - 9);
		int rs = 0;
		for (int j = 0; j < 10; j++) {
			rs += mm.x[0][j] * (9 - j);
			rs = rs % m;
		}
		if (k < 10) {
			cout << k << endl;
		}
		else {
			cout << rs << endl;
		}

	}
	return 0;
}