[蓝桥杯][基础练习VIP]矩阵乘法

问题 1472: [蓝桥杯][基础练习VIP]矩阵乘法

时间限制: 1Sec 内存限制: 128MB 提交: 1598 解决: 485

题目描述
给定一个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<bits/stdc++.h>
using namespace std;
int a[35][35]={0};//结果矩阵
int b[35][35]={0};//初始矩阵
int c[35][35]={0};//中间矩阵
int main()
{
    int n,m;
    cin>>n>>m;
    if(n==0){cout<<0;return 0;}//矩阵为0
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
        cin>>b[i][j];
        c[i][j]=b[i][j];
        }

    if(m==0)//0阶
    {
        for(int i=1;i<=n;i++)
        {
        for(int j=1;j<=n;j++)
        {
            if(i==j)cout<<1<<" ";
            else cout<<0<<" ";
        }
        cout<<endl;
        }
        return 0;
    }

     for(int i=1;i<m;i++)
     {
         for(int d=1;d<=n;d++)
           for(int e=1;e<=n;e++)
            {
                a[d][e]=0;
            }//加之前要把a[][]清零！
         for(int l=1;l<=n;l++)
          for(int p=1;p<=n;p++)
          {
              for(int j=1;j<=n;j++)
               {
                 a[l][p]+=c[l][j]*b[j][p];
               }
          }//计算，每次c的行不变，b的列不变
          for(int d=1;d<=n;d++)
           for(int e=1;e<=n;e++)
            {
                c[d][e]=a[d][e];
            }//保留上次的计算结果
     }
     for(int i=1;i<=n;i++)
     {
        for(int j=1;j<=n;j++)
        {
            cout<<a[i][j]<<" ";
        }
        cout<<endl;
     }
     return 0;
}