矩阵乘法

矩阵乘法

2021春季学期面向对象程序设计第五、六周课外练习

描述

写一个Matrix类，使得下面程序能够实现矩阵乘法

程序填空

#include <iostream>
#include <cstring>
using namespace std;

class Matrix {
public:
    int ** p;
// 在此处补充你的代码
friend ostream & operator << (ostream &o, const Matrix &a){
        for(int i = 0;i < a.r;i ++){
            for(int j = 0;j < a.c;j ++) o << a.p[i][j] << " ";
            o << endl;
        }
        return o;
    }
};

int main() {
    Matrix a, b;
    cin >> a >> b;
    cout << a * b << endl;
    return 0;
}

输入

用空行隔开的两个矩阵。矩阵第一行是行、列数。接下来是矩阵的内容。

输出

一个矩阵

输入样例

3 2
1 2
3 4
5 6

2 2
7 8
9 0

输出样例

25 8
57 24
89 40

解析

根据主函数可以看到，我们需要做两个事情：1.重载输入流符号>>。2.重载乘法符号*。

1. 重载输入流符号>>

   int r, c; // 定义行、列数
   friend istream &operator>>(istream &in, Matrix &a)
   {
   
       cin >> a.r >> a.c; // 输入行、列数
   
       a.p = new int *[a.r];
       for (int i = 0; i < a.r; i++) // 按行输入
       {
           a.p[i] = new int[a.c];
           for (int j = 0; j < a.c; j++) //按列输入
           {
               in >> a.p[i][j];
           }
       }
       return in;
   }
   

2. 重载乘法符号*
   这里需要注意一点，在主函数当中，乘法符号*前后2个变量都是Matrix类型的变量，所以要用友元函数来实现。通过题目已经重载好的输出流函数可以知道，我们要返回的类型是Matrix类型。剩下的就是通过编程实现矩阵乘法的逻辑了。

   friend Matrix operator*(Matrix &a, Matrix &b)
   {
       Matrix result;
       result.r = a.r;
       result.c = b.c;
       result.p = new int *[a.r];
       for (int ai = 0; ai < a.r; ai++) // 遍历a的行
       {
           result.p[ai] = new int[b.c];
           for (int bj = 0; bj < b.c; bj++) //遍历b的列遍
           {
               int tmp = 0;
               for (int aj = 0; aj < a.c; aj++) // 遍历a的列,同时也是b的行
               {
                   tmp += a.p[ai][aj] * b.p[aj][bj];
               }
               result.p[ai][bj] = tmp;
           }
       }
       return result;
   }
   

完整代码

#include <iostream>
#include <cstring>
using namespace std;

class Matrix
{
public:
    int **p;
    // 在此处补充你的代码
    int r, c; // 定义行、列数
    friend istream &operator>>(istream &in, Matrix &a)
    {

        cin >> a.r >> a.c; // 输入行、列数

        a.p = new int *[a.r];
        for (int i = 0; i < a.r; i++) // 按行输入
        {
            a.p[i] = new int[a.c];
            for (int j = 0; j < a.c; j++) //按列输入
            {
                in >> a.p[i][j];
            }
        }
        return in;
    }
    friend Matrix operator*(Matrix &a, Matrix &b)
    {
        Matrix result;
        result.r = a.r;
        result.c = b.c;
        result.p = new int *[a.r];
        for (int ai = 0; ai < a.r; ai++) // 遍历a的行
        {
            result.p[ai] = new int[b.c];
            for (int bj = 0; bj < b.c; bj++) //遍历b的列遍
            {
                int tmp = 0;
                for (int aj = 0; aj < a.c; aj++) // 遍历a的列,同时也是b的行
                {
                    tmp += a.p[ai][aj] * b.p[aj][bj];
                }
                result.p[ai][bj] = tmp;
            }
        }
        return result;
    }

    friend ostream &operator<<(ostream &o, const Matrix &a)
    {
        for (int i = 0; i < a.r; i++)
        {
            for (int j = 0; j < a.c; j++)
                o << a.p[i][j] << " ";
            o << endl;
        }
        return o;
    }
};

int main()
{
    Matrix a, b;
    cin >> a >> b;
    cout << a * b << endl;
    return 0;
}