A1041. 矩阵乘法

A1041. 矩阵乘法

时间限制： 1.0s   内存限制： 512.0MB  

问题描述

　　给定一个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

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <string.h>
5. int main(){
6.     int n,m;
7.     while((scanf("%d %d",&n,&m))!=EOF){
8.         int a[30][30]={0},tmp[30][30]={0},ans[30][30]={0};
9.         for(int i=0;i<n;i++){
10.             for(int j=0;j<n;j++){
11.                 scanf("%d ",&a[i][j]);
12.                 tmp[i][j]=a[i][j];
13.                 ans[i][j]=a[i][j];
14.             }
15.         }
16.         if(m==0){
17.             for(int i=0;i<n;i++){
18.                 for(int j=0;j<n;j++){
19.                     if(i==j)
20.                         printf("1 ");
21.                     else
22.                         printf("0 ");
23.                 }
24.                 printf("\n");
25.             }
26.         }
27.         else{
28.             while(m>1){              
29.                  for(int i=0;i<n;i++){  
30.                      for(int j=0;j<n;j++){  
31.                          tmp[i][j]=ans[i][j];  
32.                      }  
33.                  }
34.                  memset(ans,0,sizeof(ans));
35.                  for(int i=0;i<n;i++){
36.                      for(int j=0;j<n;j++){
37.                          for(int k=0;k<n;k++){
38.                              ans[i][j]+=a[i][k]*tmp[k][j];
39.                          }
40.                      }
41.                  }
42.                  m--;
43.             }
44.             for(int i=0;i<n;i++){
45.                 for(int j=0;j<n;j++){
46.                     printf("%d ",ans[i][j]);
47.                 }
48.                 printf("\n");
49.             }
50.         }
51.     }
53.     system("pause");
54.     return 0;
55. }