基础练习 矩阵乘法

问题描述

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

第一次提交

#include <iostream>
#include<cstring>
#include<stdio.h>
using namespace std;
//矩阵乘法
int a[31][31];
int b[31][31];
int c[31][31];
void Fact(int m,int n)
{
    for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++)
        c[i][j]=a[i][j];//中间值
    memset(b,0,sizeof(b));
    //n次幂t得循环n-1次
    for(int t=1;t<n;t++){
    for(int i=1;i<=m;i++)
            for(int j=1;j<=m;j++)
            for(int k=1;k<=m;k++)
            {b[i][j]=b[i][j]+a[i][k]*c[k][j];}
    //
      for(int i=1;i<=m;i++)
        for(int j=1;j<=m;j++)
        c[i][j]=b[i][j];
        memset(b,0,sizeof(b));
    }
}
int main()
{
    int m;
    int n;
    cin>>m>>n;
    //求m阶矩阵的n次幂

    for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++)
    cin>>a[i][j];
    Fact(m,n);
    for(int i=1;i<=m;i++)
    {for(int j=1;j<=m;j++)
        cout<<c[i][j]<<" ";
    cout<<endl;}
}

错误样例输入：
10 0
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
10 10 10 10 10 10 10 10 10 10
错误样例输出：
1 0 0 0 0 0 0 0 0 0
0 1 0 0 0 0 0 0 0 0
0 0 1 0 0 0 0 0 0 0
0 0 0 1 0 0 0 0 0 0
0 0 0 0 1 0 0 0 0 0
0 0 0 0 0 1 0 0 0 0
0 0 0 0 0 0 1 0 0 0
0 0 0 0 0 0 0 1 0 0
0 0 0 0 0 0 0 0 1 0
0 0 0 0 0 0 0 0 0 1

所以我少考虑了一种情况
矩阵的零次幂
在线性代数中，矩阵的0次幂被定义为与它同维度的单位矩阵

所以函数做如下修改：

第二次提交

#include <iostream>
#include<cstring>
#include<stdio.h>
using namespace std;
//矩阵乘法
int a[31][31];
int b[31][31];
int c[31][31];
void Fact(int m,int n)
{
    for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++)
        c[i][j]=a[i][j];//中间值
    memset(b,0,sizeof(b));
    //n次幂t得循环n-1次
    for(int t=1;t<n;t++){
    for(int i=1;i<=m;i++)
            for(int j=1;j<=m;j++)
            for(int k=1;k<=m;k++)
            {b[i][j]=b[i][j]+a[i][k]*c[k][j];}
    //
      for(int i=1;i<=m;i++)
        for(int j=1;j<=m;j++)
        c[i][j]=b[i][j];
        memset(b,0,sizeof(b));//b[][]每次用来保存中间值
    }
}
int main()
{
    int m;
    int n;
    cin>>m>>n;
    //求m阶矩阵的n次幂

    for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++)
    cin>>a[i][j];
    if(n==0)//如果输出矩阵的0次幂,输出单位矩阵
    {
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++){
            if(i==j)cout<<1<<" ";
            else cout<<0<<" ";
        }
        cout<<endl;
        }
        return 0;
    }
    //如果n!=0,执行Fact()函数
    Fact(m,n);
    //输出结果c[][]
    for(int i=1;i<=m;i++)
    {for(int j=1;j<=m;j++)
        cout<<c[i][j]<<" ";
    cout<<endl;}
    return 0;
}