1141 BASIC-17 矩阵乘法

题目描述

　给定一个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>
int main()
{
	int N,M,j,i,k;
	int b[30][30],a[30][30],t[30][30];
	while(~scanf("%d %d",&N,&M))
	{
		for(i=0;i<N;i++)
		{
		  for(j=0;j<N;j++)
		  {
		     scanf("%d",&b[i][j]);
		     a[i][j]=b[i][j];
		  }
	    }
	    if(M==0)
	    {
	    	for(i=0;i<N;i++)
	    	{
	    	  for(j=0;j<N;j++) 
			  {
			     
	    		if(i==j)
	    		  b[i][j]=1;
	    		else
	    		  b[i][j]=0;
	    	   }
			}
		}
		else
		{ 
		  while(--M)
		  {
			for(i=0;i<N;i++)
		     {
		       for(j=0;j<N;j++)
		         {
		         	for(k=0;k<N;k++)
		         	  {
		         	  	t[i][j]+=b[i][k]*a[k][j];
					  }
				  }
	          }
	         for(i=0;i<N;i++)
		        {
		          for(j=0;j<N;j++)
		            {
		              b[i][j]=t[i][j];
		              t[i][j]=0;
		            }
    	        }
            }
        }
        for(i=0;i<N;i++)
          {
          	for(j=0;j<N;j++)
          	{
          	  printf((j+1)!=N?"%d ":"%d\n",b[i][j]);	
			}
		  }
	}
	return 0;
}