前缀和与差分

前缀和

一维前缀和

• 定义: 一个下标从 1 开始的一维数组 A , 前缀和$S[n]={\sum }_{i=1}^{n}A[i]$
• 时间复杂度:
  • 预处理 $O(n)$
  • 区间查询 $O(1)$
• 使用场景: 对一个不带修改的序列做多次区间查询

模板

// 预处理
for(int i = 1; i <= n; i ++ ) s[i] = s[i - 1] + a[i];

// 查询区间[L, R]
s[R] - s[L - 1]

例题

洛谷P8218: 求区间和

AC code

#include <iostream>
using namespace std;
const int N = 1E5 + 10;

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

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i ++ ) cin >> a[i];
    for(int i = 1; i <= n; i ++ ) s[i] = s[i - 1] + a[i];
    
    cin >> m;
    while(m -- )
    {
        int l, r;
        cin >> l >> r;
        cout << s[r] - s[l - 1] << '\n';
    }
}

二维前缀和

• 定义: 一个下标从 1 开始的矩阵 A , 前缀和$S[n][m]={\sum }_{i=1}^n{\sum }_{j=1}^{m}A[i][j]$
• 时间复杂度:
  • 预处理 $O(n^2)$
  • 区间查询 $O(1)$
• 使用场景: 对一个不带修改的矩阵做多次子矩阵查询

模板

// 预处理
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];

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

例题

洛谷P1719: 最大加权矩形

AC code

#include <iostream>
using namespace std;
const int N = 150;

int n, ans = -1E9;
int a[N][N], s[N][N]; 

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= n; j ++ )
            cin >> a[i][j];
            
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= n; j ++ )
            s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
            
    for (int x1 = 1; x1 <= n; x1 ++ )
        for (int y1 = 1; y1 <= n; y1 ++ )
            for (int x2 = x1; x2 <= n; x2 ++ )
                for (int y2 = y1; y2 <= n; y2 ++ )
                {
                    int x = s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
                    ans = max(ans, x);
                }
    cout << ans;
    return 0;
}

一维差分

• 定义: 一个下标从 1 开始的数组A , 差分$B[i]=A[i]-A[i-1]$, B 的前缀和即为 A
• 时间复杂度:
  • 预处理 $O(n)$
  • 区间修改 $O(1)$
• 使用场景: 对一个数组做多次区间修改

模板

// 预处理
for(int i = 1; i <= n; i ++ ) b[i] = a[i] - a[i - 1];

// 区间修改[L, R] 加上 C
b[L] += C, b[R + 1] -= C

// 修改后的序列即b的前缀和
for(int i = 1; i <= n; i ++) b[i] += b[i - 1]

例题

洛谷P2367: 语文成绩

AC code

#include <iostream>
using namespace std;
const int N = 5E6 + 10;

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

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i ++ ) cin >> a[i];
    for(int i = 1; i <= n; i ++ ) b[i] = a[i] - a[i - 1];
    
    while(m -- )
    {
        int l, r, c;
        cin >> l >> r >> c;
        b[l] += c;
        b[r + 1] -= c;
    }
    
    for(int i = 1; i <= n; i ++ ) b[i] += b[i - 1];
    
    int ans = 1E9;
    for(int i = 1; i <= n; i ++ ) ans = min(ans, b[i]);
    cout << ans;
    return 0;
}

二维差分

• 定义: 一个下标从 1 开始的矩阵 A , 差分矩阵 B 的前缀和即为矩阵 A
• 时间复杂度:
  • 预处理 $O(n^2)$
  • 区间修改 $O(1)$
• 使用场景: 对一个矩阵做多次子矩阵修改

模板

// 预处理即给每个以(xi, yi)为左上角，(xi, yi)为右下角的子矩阵中的所有元素加上a[i][j]

//给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上c
b[x1, y1] += c, b[x2 + 1, y1] -= c, b[x1, y2 + 1] -= c, b[x2 + 1, y2 + 1] += c

// 修改后的矩阵即b的前缀和
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] + b[i][j];

例题

洛谷P3397: 地毯

AC code

#include <iostream>
using namespace std;
const int N = 1010;

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

int main()
{
    cin >> n >> m;
    while(m -- )
    {
        int x1, y1, x2, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        b[x1][y1] ++;
        b[x1][y2 + 1] --;
        b[x2 + 1][y1] --;
        b[x2 + 1][y2 + 1] ++;
    }
    
    for(int i = 1;i <= n; i ++ )
        for(int j = 1; j <= n; j ++ )
            a[i][j] = a[i - 1][j] + a[i][j - 1] + b[i][j] - a[i - 1][j - 1];
            
    for(int i = 1; i <= n; i ++ )
    {
        for(int j = 1; j <= n; j ++ )
            cout << a[i][j] << ' ';
        cout << '\n';
    }
    return 0;
}