c++ 矩阵类运算符重载

C++ 矩阵类运算符重载

运算符重载，是C++静态多态性实现方法之一——运算符重载，此题还综合了一些C++的基础知识，例如一些成员函数的实现。

#include<iostream>
#include<stdlib.h>// 用于退出程序
using namespace std;
class Matrix// 设立一个矩阵类
{
	int row,col;
	int *p;// 用来储存矩阵元素
public:
	Matrix();
	Matrix(int r,int c);
	Matrix(const Matrix &m);
	void disp();//展示矩阵
	friend Matrix operator + (const Matrix &am,const Matrix &bm);// 加法运算符的重载
	Matrix & operator = (const Matrix &a);// “=”运算符的重载
	~Matrix()
	{
		delete []p;
	}
};
Matrix::Matrix()
{
	row=0;
	col=0;
	p=NULL;
}

Matrix::Matrix(int r,int c):row(r),col(c)
{
	int i=r*c;
	p=new int[i];
	cout<<"请输入该矩阵元素："<<endl;
	for(int a=0;a<i;a++)
	{
		cin>>p[a];
	}
}

Matrix::Matrix(const Matrix &m)// 当加法运算符重载函数被调用时，会调用复制构造函数，由于数据成
{                              // 员有指针，因此为了防止析构函数调用时发生错误，设立一个复制构 
                               //   造函数
	row=m.row;
	col=m.col;
	if(!p)// 当指针未被赋值时，指针被赋予随机数，或当指针被赋值后，通过此语句指针也可以被删除调 
        delete p;// 空间,从而去使用新的空间
		
	int len=col*row;
	p=new int[len];
	for(int i=0;i<len;i++)
	{
		p[i]=m.p[i];
	}
}
void Matrix::disp()
{
	for(int i=0;i<row;i++)
	{
		cout<<'\t';
		for(int j=0;j<col;j++)
			cout<<*(p+i*col+j)<<'\t';
		cout<<endl;
	}
}
Matrix operator + (const Matrix &am,const Matrix &bm)
{
	if(am.row==bm.row&&am.col==bm.col)
	{
		int i,j;
		Matrix temp(am);
		i=am.row*am.col;
		for(j=0;j<i;j++)
		{
			temp.p[j]=am.p[j]+bm.p[j];
		}
		return temp;
	}
	else
	{
		cout<<"program terminated!"<<endl;
		exit(1);
	}
}
Matrix & Matrix::operator = (const Matrix &bm)
{
	if((row==0)&&(col==0))
	{
		if(p)
		delete p;
		row=bm.row;
		col=bm.col;
		int len=bm.row*bm.col;
		p=new int[len];
		for(int i=0;i<len;i++)
		{
			p[i]=bm.p[i];
		}
		return *this;
	}
	else if((row==bm.row)&&(col==bm.col))
	{
        for(int i=0;i<row*col;i++)
		{
			p[i]=bm.p[i];
		}
		return *this;
	}


	else if((row!=bm.row)||(col!=bm.col))
	{
		cout<<"program terminated!"<<endl;
		exit(1);
	}
}
int main()
{
	int row_a,col_a,row_b,col_b;
	cout<<"请输入am矩阵的行数和列数:"<<endl;
	cin>>row_a>>col_a;
	Matrix am(row_a,col_a);
	cout<<"请输入bm矩阵的行数和列数:"<<endl;
	cin>>row_b>>col_b;
    Matrix bm(row_b,col_b),cm;
	cout<<"am:"<<endl;
	am.disp();
	cout<<"bm:"<<endl;
	bm.disp();
	cm=am+bm;
	cout<<"cm=am+bm:"<<endl;
	cm.disp();
	am=bm;
	cout<<"am=bm:"<<endl;
	am.disp();
	return 0;
}