矩阵前缀和可视化演示

C++ 实现代码

void calculatePrefixSum(const vector<vector<int>>& matrix,
                vector<vector<int>>& prefixSum) { ←
    int rows = matrix.size();
    int cols = matrix[0].size();
    
    // 初始化第一个元素
    prefixSum[0][0] = matrix[0][0];
    
    // 计算第一行
    for (int j = 1; j < cols; j++) {
        prefixSum[0][j] = prefixSum[0][j-1] + matrix[0][j];
    }
    
    // 计算第一列
    for (int i = 1; i < rows; i++) {
        prefixSum[i][0] = prefixSum[i-1][0] + matrix[i][0];
    }
    
    // 计算其他位置
    for (int i = 1; i < rows; i++) {
        for (int j = 1; j < cols; j++) {
            prefixSum[i][j] = prefixSum[i-1][j] + prefixSum[i][j-1] 
                           - prefixSum[i-1][j-1] + matrix[i][j];
        }
    }
}