矩陣的冪

題目描述

給定一個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

#include <iostream>
#include <cstdio>
using namespace std;
int a[11][11],b[11][11],c[11][11], n, k;
void Matrix_mult(int a[11][11], int b[11][11] )
{
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            c[i][j] = 0;
            for(int k = 1; k <= n; k++)
            {
                c[i][j] += a[i][k] * b[k][j];
            }
        }
    }

    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            a[i][j] = c[i][j];
        }
    }
}
int main()
{

    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
        scanf("%d", &a[i][j]),b[i][j] = a[i][j];
    
    for(int i = 2; i <= k; i++)
        Matrix_mult(a, b);
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j < n; j++)
        {
            printf("%d ", a[i][j]);
        }
        printf("%d\n",a[i][n]);
    }
    return 0;
}