WEEK14 作业 C - Q老师的考验（必做）

C - Q老师的考验（必做）

题目描述

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

题解

本周实验课上重点讲的矩阵快速幂问题。
首先推导出f(n)的公式


按照以上规律对这个10*10的矩阵做快速幂，然后用他的第一行乘以f(0)~f(9)取和即可。
快速幂模板：

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

代码

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
 
long long k, mod;

struct Matrix {
	long long x[10][10];
	Matrix() {memset(x, 0, sizeof(x));}
	
	Matrix operator *(Matrix &b) {
		Matrix ret;
		for (int i = 0; i < 10; i++){
			for (int j = 0; j < 10; j++){
				ret.x[i][j] = 0;
				for (int k = 0; k < 10; k++){
					ret.x[i][j] += (x[i][k]*b.x[k][j])%mod;
					ret.x[i][j] %= mod;
				}
			}
				
		}			
		return ret;
	}
	
};

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

int main() {
	while (scanf("%lld%lld", &k, &mod) != EOF) 
	{
		Matrix a,b,tmp;
		for (int i = 1; i < 10; i++)
			a.x[i][i-1] = 1;
		for (int i = 0; i < 10; i++)
			scanf("%lld", &a.x[0][i]);
		if (k < 10) 
		{
			printf("%lld\n", k%mod);
			continue;
		}
		tmp = quick_pow(a, k-9);
		long long ans = 0;
		for (int i = 0; i < 10; i++)
			ans = (ans + tmp.x[0][i]*(9-i)) % mod;
		printf("%lld\n", ans);
	}
	return 0;
}