使用C++动态C风格数组创建Stochastic矩阵的问题排查

问题根源 #

你的代码中，在将列元素归一化后直接对每个元素单独四舍五入，这会导致舍入误差累加，最终列的总和不再等于1。比如，假设归一化后四个元素是0.25、0.25、0.25、0.25，若保留1位小数，四舍五入后都是0.3，总和就变成1.2；或者某些元素舍入后总和不足1，这都是单独舍入的必然结果。

另外注意：你代码里写的是round(matrix[i][j] * 10) / 10，这是保留1位小数，但你描述里要求保留2位小数，这里明显是笔误，应该改成round(matrix[i][j] * 100) / 100。

修复后的代码 #

我们调整四舍五入的逻辑：对每一列的前rows-1个元素进行四舍五入，最后一个元素用1.0减去前面所有元素的和，再做舍入处理——这样能强制保证列和严格为1，同时满足小数位数要求。

修改后的CrearMatrizEstocastica函数如下：

double** CrearMatrizEstocastica(double ** matrix,int rows, int column) {
    const int decimal_places = 2; // 设置要保留的小数位数
    const double scale = pow(10, decimal_places); // 计算缩放因子，2位小数对应100

    for (int j = 0; j < column; j++) {
        double columnsum = 0;
        // 生成随机数并计算列和
        for (int i = 0; i < rows; i++) {
            double random_num = static_cast<double>(rand()) / RAND_MAX;
            matrix[i][j] = random_num;
            columnsum += matrix[i][j];
        }
        // 先完成归一化
        for (int i = 0; i < rows; i++) {
            matrix[i][j] /= columnsum;
        }
        // 处理四舍五入，保证列和为1
        double sum_rounded = 0.0;
        for (int i = 0; i < rows - 1; i++) {
            matrix[i][j] = round(matrix[i][j] * scale) / scale;
            sum_rounded += matrix[i][j];
        }
        // 最后一个元素用1减去前面的和，再舍入
        matrix[rows-1][j] = round((1.0 - sum_rounded) * scale) / scale;
        // 极端情况处理：若前面的和舍入后超过1，强制修正
        if (sum_rounded > 1.0) {
            for (int i = 0; i < rows - 1; i++) {
                matrix[i][j] = 0.0;
            }
            matrix[rows-1][j] = 1.0;
        }
    }
    cout << "Stochastic Matrix created!" << endl;
    return matrix;
}

额外补充 #

你代码中MostrarMatrizEstocastica函数未定义，这里补充一个实现方便验证：

void MostrarMatrizEstocastica(double** matrix, int rows, int cols) {
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            printf("%.2f\t", matrix[i][j]); // 按2位小数输出
        }
        cout << endl;
    }
    // 列和验证
    cout << "\n列和验证：" << endl;
    for (int j = 0; j < cols; j++) {
        double sum = 0.0;
        for (int i = 0; i < rows; i++) {
            sum += matrix[i][j];
        }
        printf("列%d和：%.2f\n", j+1, sum);
    }
}

内容的提问来源于stack exchange，提问作者tummy