9 矩阵乘法

9 矩阵乘法

作者: Turbo时间限制: 1S章节: 基本练习（循环）

问题描述 :

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22

输入说明 :

第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
　　接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

输出说明 :

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

输入范例 :

2 2
1 2
3 4

输出范例 :

7 10
15 22

#include<stdio.h>
#include<memory.h>
int main(){
	int i,j,k,t,m,n;
	int	a[30][30],temp[30][30],mid[30][30];
	while(scanf("%d %d",&n,&m)!=EOF){
		memset(a,0,sizeof(a));
		memset(temp,0,sizeof(temp));
		memset(mid,0,sizeof(mid));
		for(i=0;i<n;i++){
			for(j=0;j<n;j++){
				if(j==n-1){
					scanf("%d",&a[i][j]);
				}else{
					scanf("%d ",&a[i][j]);
				}
				temp[i][j]=a[i][j];
			}
		}
		if(m==0){
			//单位矩阵
			memset(a,0,sizeof(a));
			i=0;a[0][0]=1;
			while(i++<n)a[i][i]=1;
		}else{
			m--;
			while(m--){
				for(i=0;i<n;i++){
					for(j=0;j<n;j++){
						t=0;
						for(k=0;k<n;k++){
							t+=a[i][k]*temp[k][j];
						}
						mid[i][j]=t;
					}
				}
				for(i=0;i<n;i++){
					for(j=0;j<n;j++){
						a[i][j]=mid[i][j];
					}
				}
			}
		}
		for(i=0;i<n;i++){
			for(j=0;j<n;j++){
				if(j==n-1){
					printf("%d\n",a[i][j]);
				}else{
					printf("%d ",a[i][j]);
				}
			}
		}
	}
	return 0;
}