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(x)={x,a0×f(x−1)+a1×f(x−2)+a2×f(x−3)+……+a9×f(x−10),if(x<10)if(x≥10)
所以，转移矩阵为:
（代码中有更详细的注释）

三、代码实现

#include<cstdio>
#include<cstring>
using namespace std;

int m=0,k=0,ans=0;

struct matrix{
	int ma[10][10];
	matrix(){
		memset(ma,0,sizeof ma);
	}
	matrix multiply(matrix& mat);
};

matrix matrix::multiply(matrix& mat){   //矩阵乘法运算
		matrix tmp;
		for(int i=0;i<10;i++){
			for(int j=0;j<10;j++){   //遍历并计算结果矩阵每个位置
				tmp.ma[i][j]=0;
				for(int k=0;k<10;k++){
					tmp.ma[i][j]+=ma[i][k]*mat.ma[k][j]%m;   //按照乘法法则运算
					tmp.ma[i][j]%=m;   //这里比较巧妙，因为输出要求余数，可以在计算的时候直接计算
				}
			}
		}
		return tmp;
} 

matrix my_pow(matrix mm,int x){   //x为要乘的次数
	matrix tt;
	for(int i=0;i<10;i++){   //初始化
		tt.ma[i][i]=1;
	}	
	while(x!=0){
		if(x&1){tt=tt.multiply(mm);	} 		//如果剩下的次数为1，只需要再乘一次
		mm=mm.multiply(mm);   
		x>>=1;   //每次乘法都是1次方相乘、2次方相乘、3次方相乘，所以每次计算之后要除2
	}
	return tt;
}

int main(){
	while(scanf("%d %d",&k,&m)!=EOF){
		matrix t;
		for(int i=0;i<10;i++){
			scanf("%d",&t.ma[0][i]);
			t.ma[i+1][i]=1;
		}
		if(k<10){
			printf("%d\n",k%m);	continue;
		}
		t=my_pow(t,k-9);   //t变为最后的结果矩阵 
		ans=0;
		for(int i=0;i<10;i++){
			ans=ans+t.ma[0][i]*(9-i)%m;
			ans=ans%m;
		}
		printf("%d\n",ans);
	}
	return 0;
}