题目描述

给定一个n*n的矩阵，求该矩阵的k次幂，即P^k。

---

输入描述:

第一行：两个整数n（2<=n<=10）、k（1<=k<=5），两个数字之间用一个空格隔开，含义如上所示。
接下来有n行，每行n个正整数，其中，第i行第j个整数表示矩阵中第i行第j列的矩阵元素Pij且（0<=Pij<=10）。另外，数据保证最后结果不会超过10^8。

---

输出描述:

对于每组测试数据，输出其结果。格式为：
n行n列个整数，每行数之间用空格隔开，注意，每行最后一个数后面不应该有多余的空格。

---

示例1

输入

2 2
9 8
9 3
输出
153 96
108 81

---

分析

矩阵相乘，第i行乘以j列

---

代码

#include <iostream>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

/**
* 矩阵相乘，第a行乘以b列
*/
int solut(int tem[10][10], int num[10][10], int a, int b, int n) {
    int ans = 0;
    for(int i = 0; i < n; i ++) {
        ans += tem[a][i]*num[i][b];
    }
    return ans;
}

int main() {
    int n, m,num[10][10],tem[10][10],ans[10][10];
    cin>>n>>m;
    for(int i = 0; i < n; i ++)
        for(int j = 0; j < n; j++) {
            cin>>num[i][j];
            ans[i][j] = num[i][j];
        }
	//计算
    for(int i = 1; i < m; i ++) {
        for(int j = 0; j < n; j ++){
            for(int k = 0;k < n; k ++){
               tem[j][k] = ans[j][k];
            }
        }
        //矩阵相乘
        for(int j = 0; j < n; j ++) {
            for(int k = 0; k < n; k ++) {
                ans[j][k] = solut(tem,num,j,k,n);
            }
        }
    }
	//输出
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            cout<<ans[i][j];
            if(j==n-1) {
                cout<<endl;
            } else {
                cout<<" ";
            }
        }
    }
    return 0;
}