前缀和与差分

前缀和与差分

一、前缀和

 

1.1 一维前缀和

一维前缀和是解决连续区间和查询问题的有效技巧，给定长度为n的数组，对于给定的m个询问，每个询问给出一个区间[l, r]，求区间内元素的总和。

$1\le n,m\le 100000$

$1\le l\le r\le n$

朴素法：对于每次询问，遍历区间内元素并加和

时间复杂度：O(n^2)

前缀和法：预处理，当前项加上前面所有项

a0 = 0

a1 = a1+ a0

a2 = a1 + a2

a3 = a2 + a3

…

对于每次查询，只需要查表即可，对于[l, r]区间和，等于：a[r] - a[l-1]

因为

a[r] = a[0] + a[1] + a[2] + … + a[r];

a[l-1] = a[0] + a[1] + … + a[l-1];

所以

a[r] - a[l-1] = a[l] + a[l+1] + … + a[r];

代码如下：

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;
const int N = 100010;
int n, m, s[N];

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

---

1.2 二维前缀和

二维前缀和是在一维的基础上，是为了解决多次连续求二维区间和。

输入一个 n 行 m 列的整数矩阵，再输入 q 个询问，每个询问包含四个整数 x1,y1,x2,y2表示一个子矩阵的左上角坐标和右下角坐标。对于每个询问输出子矩阵中所有数的和。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1010;

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

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

---

二、差分

 

2.1 一维差分

一维差分是解决连续区间修改操作的有效技巧，给定长度为 n 的整数序列。

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

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

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m;

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

---

2.2 二维差分

二维差分是在一维差分的基础上，是为了解决多次对指定二维区间进行修改。输入一个 n 行 m 列的整数矩阵，再输入 q 个操作，每个操作包含五个整数 x1,y1,x2,y2,c其中 (x1,y1)和 (x2,y2)表示一个子矩阵的左上角坐标和右下角坐标。每个操作都要将选中的子矩阵中的每个元素的值加上 c。

请你将进行完所有操作后的矩阵输出。

代码：

#include <iostream>
#include <cstring>

using namespace std;

int a[1010][1010];
int n, m, q;

void insert(int x1, int y1, int x2, int y2, int c)//这个是核心
{
    a[x1][y1] += c;
    a[x1][y2+1] -= c;
    a[x2+1][y1] -= c;
    a[x2+1][y2+1] += c;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> n >> m >> q;
    int x = 0;
    for (int i = 1; i <= n; ++ i)
    {
        for (int j = 1; j <= m; ++ j)
        {
            cin >> x;
            insert(i, j, i, j, x);
        }
    }
    int x1, y1, x2, y2, c;
    while (q --)
    {
        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][j] + a[i-1][j] + a[i][j-1] - a[i-1][j-1];
            cout << a[i][j] << ' ';
        }
        cout << "\n";
    }
    
    return 0;
}

---

THEEND…