一维、二维前缀和

目录

一、转换为一维前缀和问题

        1、一维前缀和

        2、二维转一维后求解

二、二维前缀和

---

一、转换为一维前缀和问题

        1、一维前缀和

                数据数组nums和初试话一个presum数组。presum多一列，避免处理特殊情况i=0。

for (int i = 1; i < preSum.size(); i++) {
                preSum[i] = preSum[i - 1] + nums[i - 1];
            }

                通过求下标left到right的数组和：

int sumRange(int left, int right) {
            return preSum[right + 1] - preSum[left];
        }

        2、二维转一维后求解

                 martix数组为M行N列，则创建M行，N+1列二维数组presum，presum每一行都是martix每一行的一维前缀和。

typedef struct {
    int **presum;
} NumMatrix;

NumMatrix* numMatrixCreate(int** matrix, int matrixSize, int* matrixColSize) {
    NumMatrix *ret=(NumMatrix*)malloc(sizeof(NumMatrix));
    ret->presum=(int**)malloc(sizeof(int*)*matrixSize);
    if(matrixSize==0|| matrixColSize[0]==0)
    {
        ret->presum=NULL;
        return ret;
    }
     for(int i=0;i<matrixSize;i++)
    {  
        ret->presum[i]=(int*)malloc(sizeof(int)*(matrixColSize[i]+1));
        ret->presum[i][0]=0;

        for(int j=1;j<=matrixColSize[i];j++)
        {
            ret->presum[i][j]=matrix[i][j-1]+ret->presum[i][j-1];
        }
    }
    return ret;
}
int numMatrixSumRegion(NumMatrix* obj, int row1, int col1, int row2, int col2) {
    int sum=0;
    for (int i = row1; i <= row2; i++)
    {
        sum+=obj->presum[i][col2+1]-obj->presum[i][col1];
    }
    return sum;
}
void numMatrixFree(NumMatrix* obj) {
    free(obj);
}

二、二维前缀和

        声明：蓝色区域是数据数组matrix，整个数组是定义的预处理数组presum，presum数组比matrix数组多一行一列(灰色区域，以下会讲到为何多一行一列)。presum[i][j]保存的是黄色区域数值的总和。

        1、初试话presum数组

        首先我们要明确怎么得到presum[i][j]的值。以下绿色区域值的总和分别是presum[i-1][j]、presum[i][j-1]的值。黑色区域是presum[i-1][j-1]的值。所以

presum[i][j]=presum[i-1][j]+presum[i][j-1]-presum[i-1][j-1]+matrix[i-1][j-1]

所以i、j从1开始遍历，不然对于i、j=0时要特殊单独处理。

         2、根据presum得坐标(x1,y1,x2,y2)的值

在presum中，第i行第j列的数组值是martix数组前i-1行j-1列的元素和。

坐标是基于martix数组而来的，所以在presum中，横纵坐标都的+1，比如，计算(0,0,x2,y2)，其值为presum[x2+1][y2+1]。

坐标值(x1,y1,x2,y2)=presum[x2+1][y2+1]-presum[x1][y2+1]-presum[x2+1][y1]+presum[x1][y1]

typedef struct {
    int **presum;
    int s;
} NumMatrix;
NumMatrix* numMatrixCreate(int** matrix, int matrixSize, int* matrixColSize) {

    NumMatrix *ret=(NumMatrix*)malloc(sizeof(NumMatrix));
    ret->presum=(int **)malloc(sizeof(int*)*(matrixSize+1));
    ret->s = matrixSize + 1;
     if(matrixSize==0|| matrixColSize[i]==0)
    {
        ret->presum=NULL;
        return ret;
    }

    for(int i=0;i<=matrixSize;i++)
    {
        ret->presum[i]=(int*)malloc(sizeof(int)*(matrixColSize[i]+1));
        for(int j=0;j<=matrixColSize[0];j++)
        ret->presum[i][j]=0;

    }
    
    for(int i=1;i<=matrixSize;i++)
    {
        for(int j=1;j<=matrixColSize[0];j++)
        {
         ret->presum[i][j] = ret->presum[i - 1][j] + ret->presum[i][j - 1] + matrix[i - 1][j - 1] - ret->presum[i - 1][j - 1];
        }
    }
    return ret;
}
int numMatrixSumRegion(NumMatrix* obj, int row1, int col1, int row2, int col2) {
    return (obj->presum[row2 + 1][col2 + 1] - obj->presum[row1][col2 + 1] - obj->presum[row2 + 1][col1] + obj->presum[row1][col1]);
}
void numMatrixFree(NumMatrix* obj) {
     for (int i = 0; i < obj->s; i++) {
        free(obj->presum[i]);
    }
    free(obj->presum);
}