[HDU] A Simple Math Problem

矩阵快速幂裸题,可以当做模板题练手。


#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cmath>

using namespace std;
const int N = 15;

int m;

struct matrix{
	int M[N][N];
	int l,r;
	
	void operator * (matrix x){
		int i,j,k; matrix t;
		for(i=1;i<=l;i++)
			for(j=1;j<=r;j++) t.M[i][j]=0;
		for(i=1;i<=r;i++)
			for(j=1;j<=x.l;j++)
				for(k=1;k<=l;k++)
					t.M[j][i]=(t.M[j][i]+M[k][i]*x.M[j][k]) % m;
		for(i=1;i<=l;i++)
			for(j=1;j<=r;j++) M[i][j]=t.M[i][j];
	}
		
}c,res;

int main(){
	int i,j,n,k;
	while(scanf("%d%d",&k,&m)!=EOF){
		for(i=10;i>=1;i--)
			scanf("%d",&c.M[10][i]);
		for(i=1;i<=9;i++)
			for(j=1;j<=10;j++)
				c.M[i][j]=((j-1)==i);
		c.l=10; c.r=10;
		/*for(i=1;i<=10;i++){
			for(j=1;j<=10;j++) printf("%d",c.M[i][j]);
			cout<<endl;
		}*/
		for(i=0;i<=9;i++) res.M[i+1][1]=i;
		res.l=10; res.r=1;
		for(; k ; k >>= 1){
			if(k & 1){
				res * c;
			}
			c * c;
		}
		printf("%d\n",res.M[1][1]);
	}
	return 0;
}

posted @ 2017-12-09 09:47  FranceDisco  阅读(114)  评论(0编辑  收藏  举报