温馨提示×

C++中如何自定义矩阵类模板

c++
小樊
84
2024-08-29 22:25:31
栏目: 编程语言

在C++中,你可以通过创建一个类模板来实现自定义矩阵类。以下是一个简单的示例:

#include<iostream>
#include<vector>

template<typename T>
class Matrix {
public:
    // 构造函数
    Matrix(int rows, int cols) : rows_(rows), cols_(cols) {
        data_.resize(rows);
        for (int i = 0; i< rows; ++i) {
            data_[i].resize(cols);
        }
    }

    // 获取矩阵的行数
    int rows() const {
        return rows_;
    }

    // 获取矩阵的列数
    int cols() const {
        return cols_;
    }

    // 获取矩阵中指定位置的元素
    T& operator()(int row, int col) {
        return data_[row][col];
    }

    // 获取矩阵中指定位置的元素(常量版本)
    const T& operator()(int row, int col) const {
        return data_[row][col];
    }

private:
    int rows_;
    int cols_;
    std::vector<std::vector<T>> data_;
};

int main() {
    // 创建一个3x3的整数矩阵
    Matrix<int> mat(3, 3);

    // 初始化矩阵
    for (int i = 0; i < mat.rows(); ++i) {
        for (int j = 0; j < mat.cols(); ++j) {
            mat(i, j) = i * j;
        }
    }

    // 输出矩阵
    for (int i = 0; i < mat.rows(); ++i) {
        for (int j = 0; j < mat.cols(); ++j) {
            std::cout<< mat(i, j) << " ";
        }
        std::cout<< std::endl;
    }

    return 0;
}

这个示例中,我们创建了一个名为Matrix的类模板,它接受一个类型参数T。这个类模板包含一个二维向量data_,用于存储矩阵的数据。我们还提供了一些成员函数,如rows()cols()operator(),以方便地访问和操作矩阵的元素。

main()函数中,我们创建了一个3x3的整数矩阵,并使用嵌套循环初始化和输出它。你可以根据需要修改这个类模板,以添加更多的功能和操作。

0