【Week14作业 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老师 的考验吗？

输出文件包含多组测试用例，每组测试用例格式如下：
第一行给定两个正整数 k m。（k < 2e9, m < 1e5）
第二行给定十个整数，分别表示 a0～a9。

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

---

思路：

通过矩阵乘法的定义可以构造出以上转移矩阵。然后套用矩阵快速幂的板子（乘法后要对m取余）就可以计算出中间的n-9次矩阵，最后答案就是f(k)。

---

总结：

一道矩阵快速幂来求解线性递推的题目，构造转移矩阵是关键。同时也要理解矩阵快速幂的思想。

---

代码：

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

int a[10],k,m;
const int N=10;
const int f[10]={0,1,2,3,4,5,6,7,8,9};
struct Matrix
{
	int x[N][N];
	Matrix operator * (const Matrix &t)	const
	{
		Matrix ret;
		for(int i=0;i<N;i++)
		{
			for(int j=0;j<N;j++)
			{
				ret.x[i][j]=0;
				for(int k=0;k<N;k++)
				{
					ret.x[i][j]+=x[i][k]*t.x[k][j];
					ret.x[i][j]%=m;
				}
			}
		}
		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;
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			ret.x[i][j]=0;
	for(int i=0;i<N;i++)
		ret.x[i][i]=1;
	while(x)
	{
		if(x&1)
			ret=ret*a;
		a=a*a;
		x>>=1;
	}
	return ret;
}
int main()
{
	while(cin>>k>>m)
	{
		for(int i=0;i<10;i++)
			cin>>a[i];
		Matrix ma;
		for(int i=0;i<10;i++)
			ma.x[0][i]=a[i];
		for(int i=1;i<10;i++)
			ma.x[i][i-1]=1;
		if(k<10)
			cout<<f[k]%m<<endl;
		else
		{
			Matrix ans=quick_pow(ma,k-9);
			int fk=0;
			for(int i=0;i<10;i++)
				fk+=ans.x[0][i]*f[9-i],fk=fk%m;
			cout<<fk<<endl;
		}
	}
}