【算法基础】差分

主要解决的问题

1. 求某一段区间 ± 某一数值
2. 给某一二维区间 ± 某一数值

1.一维差分

1.1 算法步骤

1. 求当前数组项与前一项的差值。如上面的差值就是 1，1，1，1，1，1，1
2. 在区间 [2,5]（2和5分别是所给区间的 [L,R] ） ,这段区间加上 2，就是第二个差值 1处加上2，在到 5 + 1 上减2,（这样在后面求前缀和时候刚好可以抵消）。[2,4] 区间和这里类似
3. 在求前缀和，上篇文章有
4. 在打印出来

1.2 代码实现

#include <iostream>
using namespace std;
#include <vector>
// 用与快速求出一段区间的子区间加减C,传统是O(n*m)
// 1.给出一段数组  2.我们得出它的差分(两个数的差值存在数组) 3.对要求的区间+-c 4.求其前缀和 5.打印出来
vector<int> a(10), s(10);
int main()
{
    int n, m;
    cin >> n >> m;
    // 1.读入数组
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
    }
    // 2.求其差分数组,原数组是这前缀和
    for (int i = 1; i <= n; i++)
    {
        s[i] = a[i] - a[i - 1];
    }
    // 3对要求的区间[l,r] +- c
    while (m--)
    {
        int l, r, c;
        scanf("%d%d%d", &l, &r, &c);
        s[l] += c;
        s[r + 1] -= c;
    }
    // 4.求前缀和
    for (int i = 1; i <= n; i++)
    {
        s[i] += s[i - 1];
    }
    // 5.打印
    for (int i = 1; i <= n; i++)
    {
        cout << s[i] << " ";
    }

    return 0;
}

2.二维前缀和

2.1 算法步骤

2.2 代码实现

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
int s[N][N],a[N][N]; //前缀和数组和差分数组
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 >> s[i][j];
    //1.构建差分数组
    for (int i = 1; i <= n; i ++ )
    for (int j = 1; j <= m; j ++ )
    a[i][j]=s[i][j]+s[i-1][j-1]-s[i][j-1]-s[i-1][j];
    //2.改变差分数组的值
    while (q -- )
   {
       int x1,x2,y1,y2,c;
       cin >> x1>>y1>>x2>>y2>>c;
       a[x1][y1]+=c;
       a[x2+1][y2+1]+=c;
       a[x1][y2+1]-=c;
       a[x2+1][y1]-=c;    //x2，y2有加一，联想记忆二维
   }
   //3.更新前缀和数组
   for (int i = 1; i <= n; i ++ )
   {
    for (int j = 1; j <= m; j ++ )
    {
        s[i][j]=a[i][j]+s[i][j-1]+s[i-1][j]-s[i-1][j-1];
        cout << s[i][j]<<" ";
    }
     cout<<endl;
   }
    return 0;
}