BASIC-17 VIP试题 矩阵乘法

问题描述

　　给定一个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 <iostream>  
using namespace std;  
int b[30][30];  
int a[30][30];  
int t[30][30];

int main() {  
    int n, m;  
    cin >> n >> m;  
      
    for(int i = 0; i < n; i++) {  
        for(int j = 0; j < n; j++) {  
            cin >> a[i][j];  
            t[i][j] = a[i][j];  
        }  
    } 
//0次幂结果为单位矩阵 
    if(m == 0) {  
        for(int i = 0; i < n; i++) {  
            for(int j = 0; j < n; j++) {  
                if(i != j) {  
                    cout << 0 << " ";  
                } else {  
                    cout << 1 << " ";  
                }  
            }  
            cout << endl;  
        }  
      
    }  
//矩阵运算
    else {while(--m) {  
          for(int i = 0; i < n; i++) {  
            for(int j = 0; j < n; j++) {  
              for(int k = n; k; k--)
                   b[i][j] += t[i][k-1] * a[k-1][j];  
                 
            }  
        }  
          for(int i = 0; i < n; i++) {  
            for(int j = 0; j < n; j++) {  
                t[i][j] = b[i][j];  
                b[i][j] = 0;  
            }  
        }  
    }  
    for(int i = 0; i < n; i++) {  
        for(int j = 0; j < n; j++) {  
            cout << t[i][j] << " ";  
        }  
        cout << endl;  
    }  
 }
    return 0;  
}