1. 程式人生 > >poj3233(Matrix Power Series)解題報告

poj3233(Matrix Power Series)解題報告

add ret can 運算 -- pro eof tar a*

題目鏈接 : http://poj.org/problem?id=3233

感覺做起來包含了矩陣的所有運算了,矩陣加法,矩陣乘法,矩陣快速冪,啊。。。寫了好久

題解 : k最大10^9,太大了,但經過觀察可以折半計算,即 :

   當k是偶數,f[k] = f[k/2] * ( 1 + A^(k/2) )
    k是奇數,f[k] = f[k-1] + A^(k)

AC代碼加註釋:

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

int A[32][32],N,m,sum[32][32]={0};

void N_matrix(int (*p)[32],int (*a)[32],int (*b)[32],int N) //矩陣p=矩陣a*矩陣b 
{	int temp[32][32];
	int i,j,k;
	memset(temp,0,sizeof(temp));
	
	for(i=0;i<N;i++)
	for(j=0;j<N;j++)
	for(k=0;k<N;k++)
	temp[i][j]+=a[i][k]*b[k][j]%m;
	
	for(i=0;i<N;i++)
	for(j=0;j<N;j++)
	p[i][j]=temp[i][j]%m;
}
void N_matrixadd(int (*p)[32],int a[32][32],int b[32][32],int N) //矩陣p=矩陣a+矩陣b
{	
	
	for(int i=0;i<N;i++)
	for(int j=0;j<N;j++)
	p[i][j]=(a[i][j]+b[i][j])%m;
}
void matrix_pow(int (*A)[32],int (*p)[32],int n,int N) //矩陣快速冪 ,n次方 
{	int temp[32][32]={0},i=0,j=0;
	
	while(i<N)		//置為E矩陣 
	{	temp[i][j]=1;
		i++,j++; 
		
	}
	for(i=0;i<N;i++)
	for(j=0;j<N;j++)
	p[i][j]=A[i][j];
	
	for(;n;n/=2)
	{	if(n&1)
		{	N_matrix(temp,temp,p,N);
			
		}
		N_matrix(p,p,p,N);
	}
		for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
		p[i][j]=temp[i][j];  //將最終結果賦給p 
}

void digui(int k)
{	int temp[32][32]={0};
	
	int y[32][32]={0};
	if(k==1)
	{	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
		sum[i][j]=A[i][j];
		return;
	}
			if(k&1)
			{	
				matrix_pow(A,y,k,N);
				k--;
			}
		digui(k/2);
		
		N_matrixadd(temp,temp,sum,N);
		
		int x[32][32];
		matrix_pow(A,x,k/2,N);

		N_matrix(x,temp,x,N);
		
		N_matrixadd(temp,temp,x,N);
		
		N_matrixadd(temp,temp,y,N);
		
		for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
		sum[i][j]=temp[i][j];  //記錄每一層遞歸最終結果 
}

int main()
{	int k;
	scanf("%d%d%d",&N,&k,&m);
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
		scanf("%d",&A[i][j]);
	digui(k);
	
	for(int i=0;i<N;i++)
		{ 	printf("%d",sum[i][0]%m);
			for(int j=1;j<N;j++)
		 	printf(" %d",sum[i][j]%m);
		
			printf("\n");
		}
	return 0;
}

  

poj3233(Matrix Power Series)解題報告