hdu 1757 A Simple Math Problem 矩阵

http://acm.hdu.edu.cn/showproblem.php?pid=1757

给定表达式 求第n项：

If x < 10 f(x) = x.
If x >= 10 f(x) = a0 * f(x-1) + a1 * f(x-2) + a2 * f(x-3) + …… + a9 * f(x-10);
 

有多种构造矩阵的方法。需要注意的是我们得到一个新值时f（n），需要的是覆盖到以前f（n-1）的位置上，把f（n-1）到f（n-8）更新一下即可。

当时我想生成f（n），然后f（n-9）没用了，就用f（n）代替。发现这样就不能用矩阵快速幂了。原因是没有一般性。每次乘的矩阵不同。

法一：

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
const int N = 13;
typedef long long LL;
int m,k;
struct Matri {
	long long a[N][N];
	Matri(){
		memset(a,0,sizeof(a));
	}
}bs; 
Matri multi(Matri A,Matri B){
	int i,j,k;
	Matri tmp;
	for(i=1;i<=10;i++){
		for(j=1;j<=10;j++){
			for(k=1;k<=10;k++){
				tmp.a[i][j] = (tmp.a[i][j] + A.a[i][k]*B.a[k][j])%m;	
			}
		}
	}
	return tmp;
}
Matri pw(Matri A,int t){
	int i;
	Matri tmp;
	for(i=1;i<=10;i++)
	tmp.a[i][i] = 1;
	while(t){
		if(t&1) tmp = multi(tmp,A);
		A=multi(A,A);
		t/=2;
	}
	return tmp;
}
void prit(Matri A){
	int i,j;
	for(i=1;i<11;i++){
		for(j=1;j<11;j++){
			printf("%d ",A.a[i][j]);
		}
		puts("");
	}
	puts("");
}
int solve(){
	Matri ans = pw(bs,k-9);
 
	Matri t ;
	for(int i=1;i<=10;i++){
		t.a[1][i] = i-1;
	}
 	ans =multi(t,ans);
	return ans.a[1][10];
}

int main()
{
	int i,j,n;
	for(i=2;i<=10;i++){
		bs.a[i][i-1] = 1;
	}
	while(scanf("%d%d",&k,&m)!=EOF){
		for(i=1;i<=10;i++){
			scanf("%d",&bs.a[10-i+1][10]);
		}
		if(k<10) printf("%d\n",k);
		else {
			printf("%d\n",solve());
		}
	}
	
	return 0;
}

法二：

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
const int N = 13;
typedef long long LL;
int m,k;
struct Matri {
	long long a[N][N];
	Matri(){
		memset(a,0,sizeof(a));
	}
}bs; 
Matri multi(Matri A,Matri B){
	int i,j,k;
	Matri tmp;
	for(i=1;i<=10;i++){
		for(j=1;j<=10;j++){
			for(k=1;k<=10;k++){
				tmp.a[i][j] = (tmp.a[i][j] + A.a[i][k]*B.a[k][j])%m;	
			}
		}
	}
	return tmp;
}
Matri pw(Matri A,int t){
	int i;
	Matri tmp;
	for(i=1;i<=10;i++)
	tmp.a[i][i] = 1;
	while(t){
		if(t&1) tmp = multi(tmp,A);
		A=multi(A,A);
		t/=2;
	}
	return tmp;
}
 
int solve(){
	Matri ans = pw(bs,k-9);
	//prit(ans);
	Matri t ;
	for(int i=1;i<=10;i++){
		t.a[i][1] = 10 - i;
	}
	ans = multi(ans,t);
	//prit(ans);
	return ans.a[1][1];
}

void prit(Matri A){
	int i,j;
	for(i=1;i<11;i++){
		for(j=1;j<11;j++){
			printf("%d ",A.a[i][j]);
		}
		puts("");
	}
	puts("");
}
void text(){
	Matri t ;
	int i;
	for(int i=1;i<=10;i++){
		t.a[i][1] = 10 - i;
	}
	for(i=1;i<=k-9;i++){
		t = multi(bs,t);
		prit(t);
	}
	//ans = multi(ans,t);
}
int main()
{
	int i,j,n;
	for(i=2;i<=10;i++){
		bs.a[i][i-1] = 1;
	}
	while(scanf("%d%d",&k,&m)!=EOF){
		for(i=1;i<=10;i++){
			scanf("%d",&bs.a[1][i]);
		}
		if(k<10) printf("%d\n",k);
		else {
			text();
			//printf("%d\n",solve());
		}
	}
	
	return 0;
}