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

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老师 的考验吗？

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

解题思路
用到了刚学的矩阵快速幂
先构造矩阵数据结构
再依公式计算
Code

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

int a[10];
int ans;
struct mx{
	int x[20][20];
	mx(){}
	mx(int *a){
		for(int i=0;i<10;i++){
			x[0][i]=a[i];
		}
		for(int i = 1; i < 10; ++i){
			for(int j = 0; j < 10; ++j){
				if(i-1 == j) x[i][j] = 1;
				else x[i][j] = 0;
			}
		}
	}
	mx operator*(const mx& m){
		mx nm;
		for(int i = 0; i < 10; i++){
			for(int j = 0; j < 10; j++){
				nm.x[i][j] = 0;
				for(int k = 0; k < 10; k++){
					nm.x[i][j] += (x[i][k]*m.x[k][j]);
				}
			}
		}
		return nm;
	}
	mx operator%(int d){
		for(int i = 0; i < 10; i++){
			for(int j = 0; j < 10; j++){
				x[i][j]%=d;
			}
		}
		return *this;
	}
	
};
mx qp(mx m, int x, int mod){
	mx nm;
	for(int i = 0; i < 10; ++i){
		for(int j = 0; j < 10; ++j){
			if(i == j) nm.x[i][j] = 1;
			else nm.x[i][j] = 0;
		}
	}
	while(x){
		if(x & 1) nm = nm * m % mod;
		m = m * m % mod;
		x >>= 1;
	}
	return nm;
}
int main(){
	int k,m;
	while(cin>>k>>m){
		ans=0;
	    for(int i=0;i<10;i++){
		    cin>>a[i];
	    }
	    mx mm(a);
	    /*for(int i=0;i<10;i++){
	    	for(int j=0;j<10;j++){
	    		cout<<m.x[i][j]<<' ';
			}cout<<endl;
		}*/
	    if(k<10)cout<<k%m<<endl;
	    else{
	    	mx nm=qp(mm,k-9,m);
	    	for(int i=0;i<10;i++){
	    		ans+=nm.x[0][i]*(9-i);
			}
	    	cout<<ans%m<<endl;	
		}
	    
	}
	
	
	
}