1. 程式人生 > >【舊資料整理】c++ 定義矩陣類CMatrix,過載二維陣列運算子[][]方法

【舊資料整理】c++ 定義矩陣類CMatrix,過載二維陣列運算子[][]方法

資料時間:2009-12-02之前

分享時間:2009-12-02到百度空間,空間關閉,重新整理

內容:實現一個二維(多維同理)陣列類,對下標做安全處理,避免內在越界,行列分別保留一個冗餘位置以確保任何時候可以返回合法地址

原始碼因為最早分享的論壇關閉找不到了,重新實現一個如下:

#include <iostream>
using namespace std;

class Row
{
        int* m_data = NULL;
        int m_y;
public:
        Row(int y) : m_y(y){ m_data = new int[y+1]; }
        ~Row() { delete m_data; }
        int& operator[](int j)
        {
                if(j > m_y)
                        j = m_y;
                return m_data[j];
        }
};

class CMatrix
{
        int m_x;
        int m_y;
        Row** m_data = NULL;
public:
        CMatrix(int x, int y) : m_x(x), m_y(y)
        {
                m_data = new Row*[x+1];
                for(int i = 0; i <= m_x; i++)
                        m_data[i] = new Row(m_y);
        }
        ~CMatrix()
        {
                for(int i = 0; i <= m_x; i++)
                        delete m_data[i];
                delete m_data;
        }
        Row& operator[](int i)
        {
                if(i > m_x)
                        i = m_x;
                return *m_data[i];
        }
        void Test()
        {
                for(int i = 0; i < m_x; i++)
                        for(int j = 0; j < m_y; j++)
                                (*this)[i][j] = i*m_x+j;
        }
        void Dump()
        {
                for(int i = 0; i < m_x; i++)
                {
                        for(int j = 0; j < m_y; j++)
                                cout << (*this)[i][j] << " ";
                        cout << "(" << (*this)[i][m_y] << ")" << endl;
                }
                for(int j = 0; j <= m_y; j++)
                        cout << "(" << (*this)[m_x][j] << ")";
                cout << endl;
        }
};

int main()
{
        CMatrix cm(2, 3);
        cm.Test();
        cm[0][1] = 5;
        cm[8][9] = 5;
        cm.Dump();
        return 0;
}