二维前缀和_二维差分

一、前缀和

1. 前缀和

已知原数组A：a[1] .... a[n]

构造前缀和数组：s[1] .... s[n]

初始化前缀和数组：s[i] = s[i-1] + a[i] （i从1开始）

使用：计算a[l]~a[r]之间的和

sum = s[r] - s[l-1]

2.二维前缀和

问题一：如何初始化前缀和数组？（见图中红框）

s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j]

使用

问题二：如何求以(x1,y1)为左顶点，(x2,y2)为右顶点的内部子矩阵数的和？

sum = s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1]

二、差分

1.差分

已知原数组A：a[1] .... a[n]

构造差分数组：b[1] .... b[n]

使得 a[i] = b[1] + b[2] + .... + b[i]

使用：a[l]~a[r] 都 +c

插入操作

b[l] += c;
b[r+1] -= c;

注： 可将a数组看作全为0的数组，对其进行构造即为a[i] = 在 (i,i) 区间内 + a[i]，这个过程中可以初始化差分数组b。

例题

#include <iostream>

using namespace std;

const int N = 100010;

int a[N],b[N];

//核心操作：插入
void insert(int l, int r, int c)
{
    b[l] += c;
    b[r+1] -= c;
}

int main()
{
    int n,m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
        insert(i,i,a[i]);	//构造出差分数组
    }
    //完成插入操作
    while(m--)
    {
        int l,r,c;
        cin >> l >> r >> c;
        insert(l,r,c);
    }
    
    for(int i = 1; i <= n; i++)
    {
        //插入后对已经改变的差分数组求前缀和即可
        a[i] = a[i-1] + b[i];
        cout << a[i] << " ";
    }
    
    return 0;
}

2.二维差分

已知a数组，构造b数组，使得a是b的二维前缀和，即b为a的二维差分数组

使用

以(x1,y1)为左顶点，(x2,y2)为右顶点的内部子矩阵的每个数 + c

对于差分数组

b[x1][y1] += c;
b[x1][y2+1] -= c;
b[x2+1][y1] -= c;
b[x2+1][y2+1] += c;

注： 可将a数组看作全为0的数组，对其进行构造即为a[i][j] = (i,j)到(i,j) + a[i][j]，这个过程中可以初始化差分数组b。

例题

#include <iostream>
using namespace std;

const int N = 1010;

int a[N][N],b[N][N];

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

int main()
{
    int n,m,q;
    cin >> n >> m >> q;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
        {
            cin >> a[i][j];
            insert(i,j,i,j,a[i][j]);	//构造出b数组
        }
    while(q--)
    {
        int x1, y1, x2, y2, c;
        cin >> 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++)
        {
            a[i][j] = a[i-1][j] + a[i][j-1] - a[i-1][j-1] + b[i][j];	//求前缀和
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
    
    return 0;
}