Question

您好我正在用c ++实现矩阵类我知道有很多像opencv这样的库，但我需要自己做。

例如，如果我实现总和，我可以这样做

class Mat{
public:
    double* data;
    int rows,cols;

    Mat(int r,int c):rows(r),cols(c){
        data = new double[r*c];
    }
};

void Sum(Mat& A,Mat& B,Mat& C){
    for (int i = 0; i < A.rows*A.cols; ++i){
        C.data[i] = A.data[i]+B.data[i];
    }   
}

int main(){    
    //Allocate Matrices
    Mat A(300,300);
    Mat B(300,300);
    Mat C(300,300);

    //do the sum
    sum(A,B,C);
}

我希望得到更具可读性的东西，但不会失去效率

C = A + B

这样每次都会重新分配C，我不想那样

感谢您的时间

Answer 1

延迟计算。

class MatAccess {
    friend class Mat;
    friend class MatOpAdd;
    virtual double operator[](int index) const = 0;
};

class MatOpAdd: public MatAccess {
friend class Mat;
private:
    const MatAccess& left;
    const MatAccess& right;
    MatOpAdd(const MatAccess& left, const MatAccess& right):
        left(left), right(right) {}
    double operator[](int index) const {
        return left[index] + right[index];
    }
};

class Mat: public MatAccess{
public:
    double* data;
    int rows,cols;

    Mat(int r,int c):rows(r),cols(c){
        data = new double[r*c];
    }

    MatOpAdd operator +(const MatAccess& other) {
        return MatOpAdd(*this, other);
    }

    const Mat& operator = (const MatAccess& other) {
        for(int i = 0; i < rows*cols; ++i) {
            data[i] = other[i];
        }
        return *this;
    }
private:
    double operator[](int index) const {
        return data[index];
    }
    double& operator[](int index) {
        return data[index];
    }
};


int main(){    
    //Allocate Matrices
    Mat A(300,300);
    Mat B(300,300);
    Mat C(300,300);

    //do the sum
    C = A + B;
}

现在'+'计算将在“operator =”

中完成

我会改变的事情：

MatAccess应包含尺寸（行，列）。
Mat添加构造函数和operator=或使其不可复制
Mat::operator+和Mat::operator=检查等行，col
不再使用时删除内存或
使用std::vector进行更简单的内存管理。

在此处创建了一个更大的示例：https://gist.github.com/KoKuToru/1d23af4bbf0b2bc89893

C ++中的高效求和和赋值运算符重载

1 个答案: