算法基础课 | 前缀和与差分

解题思路

• 下标一定从1开始
• S[i] = S[i - 1] + a[i]
• S[i]的作用：可以快速地求出数组中一段数的和

代码：

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

注意

• 数组都需要初始化为0

衍生题：子矩阵的和

• 注意输出的时候是从x1到x2所有行的内容，要用循环

while(ques -- )
    {
        int x1, y1, x2, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        for (int i = x1; i <= x2; i ++ ) ans += s[i][y2] - s[i][y1 - 1];
        cout << ans << endl;
        ans = 0;
    }

差分

• 思想：差分是前缀和的逆运算
  a：前缀和
  b：差分
  $a_i=b_1+b_2+...+b_i$
  用O(1)的时间可以给原数组的一段数字加上一个值
• a数组是b数组的前缀和数组，比如对b数组的b[i]的修改，会影响到a数组中从a[i]及往后的每一个数。

代码

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

int main()
{
    int n, m;
    cin >> n >> m;
    int a[n + 1];
    
    for (int i = 1; i <= n; i ++ ) cin >> 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;
}

解题思路

• 输入a数组
• 根据a计算b数组
• 根据要求在l到r区间加上c
• 求a数组，由于a是b的前缀和，所以对b进行前缀和计算，在输出b[i]就是a数组的值！！