#ifndef MATRIX_H

#define MATRIX_H

#include <iomanip>

#include <iostream>

#include <cstdlib>

using std::endl; using std::cerr;

using std::ostream; using std::istream;

using std::setw;

class CMatrix

{

  friend ostream &operator<<(ostream&, const CMatrix&);

public:

  CMatrix(unsigned short = 3, unsigned short = 3);

  ~CMatrix();

  CMatrix& operator=(const CMatrix&);

  friend CMatrix operator+(const CMatrix &, const CMatrix &);

  void randfill(const unsigned short);

private:

  int **mptr;				//матрица элементов

  unsigned short rows;	//количество строк в матрице

  unsigned short cols;	//количество столбцов в матрице

};

typedef CMatrix matrix;

CMatrix::CMatrix(unsigned short m, unsigned short n)

{

  if(m==0) m=1;

  if(n==0) n=1;

  rows = m;

  cols = n;

  mptr = new int*[m];

  for (unsigned short i=0; i<m; i++)

    mptr[i]=new int[n];

  for (unsigned short i=0; i<m; i++)

    for (unsigned short j=0; j<n; j++)

      mptr[i][j]=0;

    return;

  }

CMatrix::~CMatrix()

{

  for (int i=0; i<rows; i++)

    delete[]mptr[i];

  delete[]mptr;

}

ostream &operator<<(ostream &output, const CMatrix &M)

{

  for (unsigned short i=0; i<M.rows; i++)

  {

    output << endl;

    for (unsigned short j=0; j<M.cols; j++)

      output << setw(4) << M.mptr[i][j];

  }

  output << endl;

  return output;

}

CMatrix& CMatrix::operator=(const CMatrix & right)

{

  if ((rows!=right.rows)||(cols!=right.cols))

  {

    cerr << "Error: assignment is not available for matrices of different sizes" << endl;

    return *this;

  }

  for (unsigned short i=0; i<rows; i++)

    for (unsigned short j=0; j<cols; j++)

      mptr[i][j]=right.mptr[i][j];

    return *this;

  }

CMatrix operator+(const CMatrix & first, const CMatrix & second)

{

  if((first.rows!=second.rows)||(first.cols!=second.cols))

  {

    cerr << "Error: addition is not available for matrices of different sizes" << endl;

    exit(1);

  }

  unsigned short m = first.rows;

  unsigned short n = first.cols;

  CMatrix M(m,n);

  for (unsigned short i=0; i<m; i++)

    for (unsigned short j=0; j<n; j++)

      M.mptr[i][j]=first.mptr[i][j]+second.mptr[i][j];

    return M;

  }

void CMatrix::randfill(const unsigned short range)

{

  for (unsigned short i=0; i<rows; i++)

    for (unsigned short j=0; j<cols; j++)

      mptr[i][j]=rand()%range;

    return;

  }

#endif

#include "stdafx.h"

#include "matrix.h"

using std::cout;

void main ()

{

  matrix A(2,2);

  A.randfill(10);

  matrix B(2,2);

  B.randfill(10);

  cout << A;

  cout << B;

  matrix C(2,2);

  C = A+B;

  cout << C;

  return;

}