Answer to Question #213027 in C++ for Hemambar

Question

Create a class Matrix to store a m x n matrix. Include necessary constructors and functions to initialize and display the matrix. Using friend function, find the dot product of two input matrices.

Answer 1

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


class Matrix {
public:
    Matrix(int m, int n);
    ~Matrix();
    void setData(double val[]);
    void print();


private:
    int m, n;
    double *values;
    double **rows;


friend
Matrix dotProduct(const Matrix& A, const Matrix& B);
};


Matrix::Matrix(int m, int n) : m(m), n(n) 
{
    values = new double[m*n];
    rows = new double*[m];


    for (int i=0; i<m; i++) {
        rows[i] = values + n*i;
    }
}


Matrix::~Matrix() {
    delete[] rows;
    delete[] values;
}


void Matrix::setData(double val[])
{
    for (int i=0; i<m*n; i++) {
        values[i] = val[i];
    }
}


void Matrix::print() 
{
    for (int i=0; i<m; i++) {
        for (int j=0; j<n; j++) {
            cout << rows[i][j] << " ";
        }
        cout << endl;
    }
}


Matrix dotProduct(const Matrix& A, const Matrix& B)
{
    if (A.n != B.m) {
        cerr << "Error: inconsistent matixes" << endl;
        exit(1);
    }


    Matrix C(A.m, B.n);


    for (int i=0; i<A.m; i++) {
        for (int j=0; j<B.n; j++) {
            C.rows[i][j] = 0.0;
            for (int k=0; k<A.n; k++) {
                C.rows[i][j] += A.rows[i][k] * B.rows[k][j];
            }
        }
    }
    return C;
}


int main()
{
    Matrix A(3, 2);
    double v1[] = {1, 2, 3, 1, 2, 3};
    A.setData(v1);
    Matrix B(2, 3);
    double v2[] = {1, 0, 0, 1, 1, 2};
    B.setData(v2);

    Matrix C = dotProduct(A, B);

    cout << "Matrix A is:" << endl;
    A.print();
    cout << endl << "Matrix B is:" << endl;
    B.print();


    cout << endl << "C = A * B is:" << endl;
    C.print();

}