蓝桥杯 基础 矩阵乘法

题目

  基础练习 矩阵乘法  

时间限制：1.0s   内存限制：512.0MB

      

问题描述

　　给定一个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

矩阵的0次幂为单位矩阵(⊙﹏⊙)

弱弱的·代码：

#include<iostream>
#include<cstring>
int a[100][100],b[100][100],c[100][100],m;
using namespace std;
void jisuan(int x1,int y1)
{
    for(int l=0;l<m;l++)
    {
         c[x1][y1]+=b[x1][l]*a[l][y1];
    }
}
int main()
{
    int n,i,j;
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    cin>>m>>n;
        for(i=0;i<m;i++)
    {
        for(int j=0;j<m;j++)
        {
            cin>>a[i][j];
            b[i][j]=a[i][j];
        }
    }
    if(n==0)
    {
        for(i=0;i<m;i++)
        {
            for(j=0;j<m;j++)
            {
                if(i==j)
                    cout<<1<<" ";
                    else
                    cout<<0<<" ";
            }
            cout<<endl;
        }

    }
    else{
    for(i=1;i<n;i++)
    {
        for(int x=0;x<=m;x++)
        {
            for(int y=0;y<=m;y++)
            {
                jisuan(x,y);
            }
        }
         if(i<n-1)
         {
        for(int h=0;h<m;h++)
        {
            for(int r=0;r<m;r++)
            {
                b[h][r]=c[h][r];
                c[h][r]=0;
            }
        }
        }
    }
    for(i=0;i<m;i++)
    {
        for(j=0;j<m;j++)
            cout<<c[i][j]<<" ";
             cout<<endl;
    }
    }


    return 0;
}