如何在C++中从输入流中读取矩阵？

#include <iostream>
#include <vector>

using namespace std;

vector<vector<int>> readMatrixFromInputStream(istream& input) {
    int rows, cols;
    input >> rows >> cols;  // 读取矩阵的行数和列数

    vector<vector<int>> matrix(rows, vector<int>(cols));  // 创建一个二维向量来存储矩阵

    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            input >> matrix[i][j];  // 逐个读取矩阵元素
        }
    }

    return matrix;
}

int main() {
    // 从标准输入流中读取矩阵
    vector<vector<int>> matrix = readMatrixFromInputStream(cin);

    // 打印读取到的矩阵
    for (const auto& row : matrix) {
        for (const auto& element : row) {
            cout << element << " ";
        }
        cout << endl;
    }

    return 0;
}