前缀和与差分（复习复试自用）

前缀和与差分（复习复试自用）

前缀和定义：有一段数据是a1, a2, a3, …
前缀和Si=a1+a2+a3+…+ai（前缀和下标从1开始）
有了前缀和之后，计算一段和不再是O(n)的复杂度，而是一次计算，比如算[l, r]区间的和，是等于Sr - Sl-1

求Si的方法： Si = Si-1 + ai

一维前缀和模板：

#include<iostream>
using namespace std;

const int N = 1e6+10;
int n, m;
int a[N], s[N];

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];
    while(m--)
    {
        int l, r;
        cin >> l >> r;
        printf("%d\n", s[r] - s[l-1]);
    }
}

二维前缀和：

Si,j表示(i,j)左上角的元素之和，此时，
(x1, y1)和(x2, y2)之间矩形的元素之和为：
Sx2,y2 - Sx2,y1-1 - Sx1-1,y2 + Sx1-1, y1-1

二维前缀和模板

#include<iostream>
const int N = 10010;
int n, m, q;
int a[N][N], s[N][N];
int main()
{
    cin >> 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, x2, y1, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        cout << s[x2][y2] - s[x1-1][y2] - s[x2][y1-1] + s[x1-1][y1-1] << endl;
    }
    return 0;
}

差分

已知a数组:a1, a2, a3, … 要构造b数组，使得b数组是a数组的前缀和，即：ai = b1 + b2 + … + bi，b数组就是a数组的差分，a数组是b数组的前缀和，差分可以使得比如“对a数组的[l,r]区间内的数全部加c，通过差分可以将时间复杂度从o（n）降到o（1）”，即对bl + c，那么al ，al+1, … an都会加c，同时对br+1 - c，这样可以使得r后面的再减去c相当于没有变化。

差分模板题

输入一个长度为 n 的整数序列。

接下来输入 m 个操作，每个操作包含三个整数 l,r,c，表示将序列中 [l,r] 之间的每个数加上 c。

请你输出进行完所有操作后的序列。

输入格式
第一行包含两个整数 n 和 m。

第二行包含 n 个整数，表示整数序列。

接下来 m 行，每行包含三个整数 l，r，c，表示一个操作。

输出格式
共一行，包含 n 个整数，表示最终序列。

数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000
输入样例：
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例：
3 4 5 3 4 2

代码：

#include<iostream>
using namespace std;
const int N = 1e6+10;
int a[N], b[N];

void insert(int l, int r, int c)
{
    b[l] += c;
    b[r+1] -= c;
}
int main()
{
    cin >> n >> m;
    for(int i=1; i<=n; i++) scanf("%d", &a[i]);
    for(int i=1; i<=n; 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++) b[i] += b[i-1];
    for(int i=1; i<=n; i++) cout << b[i] << " ";
    return 0;
}