在C++中,你可以通过创建一个类模板来实现自定义矩阵类。以下是一个简单的示例:
#include
#include
template
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> data_;
};
int main() {
// 创建一个3x3的整数矩阵
Matrix 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的整数矩阵,并使用嵌套循环初始化和输出它。你可以根据需要修改这个类模板,以添加更多的功能和操作。