【算法模板】前缀和

前缀和的算法目的是仅用O(1) 的时间复杂度进行求数组一段区间的和

一、一维

例如求[L,R]这段区间的元素和，利用S(r)-S(l-1)这个思想实现，避免了循环递归的O(n)

#include<iostream>

using namespace std;

const int N = 1e5+10;
int S[N],a[N];
int n,m;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)        scanf("%d",&a[i]);  
    
    for(int i=1;i<=n;i++)        S[i] = S[i-1] + a[i];//前缀和的初始化，不用管S[0],因为最后计算就减掉了
    
    while(m--)
    {   int l,r;
        scanf("%d%d",&l,&r);
        printf("%d\n",S[r]-S[l-1]); //区间和的计算
    }
    
    return 0;
}

二、二维

记住下面的公式：

//1. 求s[i][j]的公式
s[i][j] = s[i - 1][j] + s[i][j - 1] -s[i - 1][j - 1] + a[i][j];

//2. 求一点 A(x1,y1)与 B(x2,y2)  这个子矩阵的和
Sab = s[x2][y2] -s[x1 - 1] [y2] - s[x2][y1 - 1] +s[x1 - 1][y1 - 1]

1、S[i, j] = 第i行j列格子左上部分所有元素的和

紫色面积是指(1, 1)左上角到(i, j - 1)右下角的矩形面积, 绿色面积是指(1, 1)左上角到(i - 1, j )右下角的矩形面积。每一个颜色的矩形面积都代表了它所包围元素的和。

从图中我们很容易看出，整个外围蓝色矩形面积s[i][j] = 绿色面积s[i - 1][j] + 紫色面积s[i][j - 1] - 重复加的红色的面积s[i - 1][j - 1] + 小方块的面积a[i][j];

2、以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和

紫色面积是指 (1, 1)左上角到(x1 - 1, y2)右下角的矩形面积 ，黄色面积是指(1, 1)左上角到(x2, y1 - 1)右下角的矩形面积；

绿色矩形的面积 = 整个外围面积s[x2, y2] - 黄色面积s[x2, y1 - 1] - 紫色面积s[x1 - 1, y2] + 重复减去的红色面积 s[x1 - 1, y1 - 1]

注意：矩阵的最左是（1，1）

#include<iostream>

using namespace std;

const int N =1010;

int s[N][N] ,a[N][N];

int main()
{
    int m,n,x1,x2,y1,y2,q;
    
    scanf("%d%d%d",&n,&m,&q);
    
    for(int i = 1;i <= n;i ++)
        for(int j=1 ;j <= m;j ++)
            scanf("%d",&a[i][j]);
            
    for(int i = 1;i <= n;i ++)
        for(int j  = 1;j <= m;j ++)
            s[i][j] = s[i-1][j] + s[i][j-1] -s[i-1][j-1] + a[i][j];//
             
    while(q--)//询问
    {
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        long  ret =  s[x2][y2] -s[x1 - 1] [y2] - s[x2][y1 -1 ] +s[x1 - 1][y1 - 1];//
        printf("%ld\n",ret);
    }
    
    return 0;
}