前缀和与差分

1. 一维前缀和

为了处理边界问题，统一使某段a[l~r]的和为s[l]-s[r]，（如a[1]）~a[10]的前缀和为s[10]-s[0]），原数组a下标从1开始，前缀和数组s下标从0开始，s[0]=0。

S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]

scanf比cin快，数据规模>=100万时建议用scanf。

例题

前缀和

#include <iostream>

using namespace std;

const int N = 1e6 + 10;
int n, m;
int a[N], s[N]; //在全局变量里定义的数组已经每个元素都初始化为0了，即这里包括了s[0] = 0

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]); //从1开始
    
    for (int i = 1; i <= n; i++) s[i] = s[i - 1] + a[i];
    
    while (m--)
    {
        int l, r;
        scanf("%d%d", &l, &r);
        printf("%d\n", s[r] - s[l - 1]);
    }
    
    return 0;
}

2. 二维前缀和

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

以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

注意：这里是二维的矩阵，i表示行j表示列，看成一个个格子，不要当成平面直角坐标系。

例题

子矩阵的和

#include <iostream>

using namespace std;

const int N = 1010;
int n, m, q;
int a[N][N], s[N][N]; //原数组和前缀和数组

int main()
{
    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--)
    {
        int x1, y1, x2, y2;
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        printf("%d\n", s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]); //算部分和
    }
    
    return 0;
}

3. 一维差分

差分是前缀和的逆运算。
原数组a[1~n]，构造数组b[1~n]，使得a[i]=b[1]+b[2]+...+b[i]，即a数组是b数组的前缀和，b数组是a数组的差分。

b1 = a1
b2 = a2 - a1
b3 = a3 - a2
...
bn = an - a(n-1)

构造差分数组

可以想成往b数组的从i到i位置插入a[i]，再把下一个数b[i-1]减去a[i]，即

void insert(int l, int r, int c)
{
	b[l] += c;
	b[r + 1] -= c;
}

for (int i = 1; i <= n; i++) insert(i, i, a[i]);

作用

如 给定区间[l, r]，让我们把a数组中的[l, r] 区间中的每一个数都加上c,即 a[l] + c , a[l + 1] + c , a[l + 2] + c , …, a[r] + c。

1. 首先让差分b数组中的 b[l] + c ,通过前缀和运算，a数组变成 a[l] + c , a[l + 1] + c, …, a[n] + c；
2. 因为r之后的元素不能影响，所有让b[r + 1] - c, 通过前缀和运算，a数组变成 a[r + 1] - c, a[r + 2] - c, …, a[n] - c。

这样可以用O(1)的时间给原数组中间的某一个区间全部加上一个值。

例题

差分

#include <iostream>

using namespace std;

const int N = 1e6 + 10;
int n, m;
int a[N], b[N]; //原数组和差分数组

void insert(int l, int r, int c)
{
    b[l] += c;
    b[r + 1] -= c;
}

int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]); //原数组a
    
    for (int i = 1; i <= n; i++) insert(i, i, a[i]); //构造差分数组b
    
    while (m--)
    {
        int l, r, c;
        scanf("%d%d%d", &l, &r, &c);
        
        insert(l, r, c);
    }
    
    for (int i = 1; i <= n; i++) b[i] += b[i - 1]; // 计算差分数组b的前缀后，即修改后的a数组
    
    for (int i = 1; i <= n; i++) printf("%d ", b[i]);
    
    return 0;
}

4. 二维差分

例题

差分矩阵

#include <iostream>

using namespace std;

const int N = 1010;
int n, m, q;
int a[N][N], b[N][N];

void insert(int x1, int y1, int x2, int y2, int c)
{
    b[x1][y1] += c;
    b[x2 + 1][y1] -= c;
    b[x1][y2 + 1] -= c;
    b[x2 + 1][y2 + 1] += c;
}

int main()
{
    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++) insert(i, j, i, j, a[i][j]); //构造差分矩阵b
    }
    
    while (q--)
    {
        int x1, y1, x2, y2, c;
        scanf("%d%d%d%d%d", &x1, &y1, &x2, &y2, &c);
        insert(x1, y1, x2, y2, c);
    }
    
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            b[i][j] += b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1]; //前缀和
        }
    }
    
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++) printf("%d ", b[i][j]);
        printf("\n");
    }
    
    return 0;
}