Advertisement

CSP考点概述:二维前缀和数组(代码实现)

阅读量:

1. 初始化扩展的二维前缀和数组

构建一个维度为 (rows + 1) x (cols + 1) 的二维前缀和矩阵,其中 rowscols 分别代表原始矩阵的行数和列数。接下来,依照特定规则对这一矩阵进行赋值操作:

复制代码
    void initPrefixSum(vector<vector<int>>& prefixSum, const vector<vector<int>>& matrix) {
    int rows = matrix.size();
    int cols = matrix[0].size();
    // 初始化扩展了的前缀和数组
    prefixSum.resize(rows + 1, vector<int>(cols + 1, 0));
    
    for (int i = 1; i <= rows; ++i) {
        for (int j = 1; j <= cols; ++j) {
            prefixSum[i][j] = matrix[i - 1][j - 1]
                            + prefixSum[i - 1][j]
                            + prefixSum[i][j - 1]

全部评论 (0)

还没有任何评论哟~