矩阵乘法

问题描述

给定一个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次幂所对应的矩阵。相邻的数之间用一个空格隔开

代码

#include <bits/stdc++.h>
int c[35][35];//保存结果
using namespace std;
int main()
{
    int a[35][35], b[35][35];
    int n, m, i, j;
    cin >> n >> m;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            cin >> a[i][j];
            b[i][j] = a[i][j];
        }
    }
    if (m == 0)
    {
        for (i = 0; i < n; i++)
        {
            for (j = 0; j < n; j++)
            {
                if (i != j)
                    cout << "0"
                         << " ";
                else
                    cout << "1"
                         << " ";
            }
            cout << endl;
        }
    }
    else
    {
        while (--m)
        {
            for (i = 0; i < n; i++)
            {

                for (j = 0; j < n; j++)
                {
                    for (int k = 0; k < n; k++)
                        c[i][j] += b[i][k] * a[k][j];
                }
            }
            for (i = 0; i < n; i++)
                for (j = 0; j < n; j++)
                {
                    a[i][j] = c[i][j];//将中间结果保存在a
                    c[i][j] = 0;//结果清零
                }
        }
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                cout << a[i][j] << " ";
            }
            cout << endl;
        }
    }
    return 0;
}