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

问题分析

直接暴力求解会超时，所以需要采用矩阵快速幂。

矩阵快速幂性质
• 快速求解线性递推式的结果，如斐波那契递推
• 优化的 DP 需要满足的条件
• 转移方程为线性递推式
• 转移次数超级多（n 很大）

对于一般的情况:
f(n) = a1f(n-1)+a2f(n-2)+…+ak*f(n-k)+b
构造矩阵：

|f(n)   |   |a1 a2 ... ak b| |f(n-1)|
|f(n-1) | = |1             | |f(n-2)|
|...    |   |    ...       | | ...  |
|f(n-k) |   |           1 0| |   1  |

很显然，这是一道需要用快速幂进行优化的题目。（不需要麻烦的求解矩阵推导公式）
矩阵的快速幂板子放在下面的代码里了，就不再这里赘述实现过程了。

#include<stdio.h>
#include<string.h>
using namespace std;
int p;
struct Matrix{
	int x[10][10];
	Matrix operator*(const Matrix&t)const{
		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]*t.x[k][j]%p;
					ret.x[i][j]%=p;
				}
			}
		return ret;
	}
	Matrix(){memset(x,0,sizeof x);}
	Matrix(const Matrix&t){memcpy(x,t.x,sizeof x);}
};
Matrix quick_pow(Matrix a,int x){
	Matrix ret;
	memset(ret.x,0,sizeof ret.x);
	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(){
	int k;
	while(scanf("%d%d",&k,&p)!=EOF){
		Matrix a;
		for(int i=0;i<9;i++)
			a.x[i+1][i]=1;
		for(int i=0;i<10;i++)
			scanf("%d",&a.x[0][i]);
		if(k>9){
			Matrix f=quick_pow(a,k-9);
			int ans=0;
			for(int i=0;i<10;i++){
				ans+=f.x[0][i]*(9-i)%p;
				ans%=p;
			}
			printf("%d\n",ans);	
		}
		else printf("%d\n",k%p);
	}
	
	return 0;
}