差分
讲解
我们先来说说前缀和。
前缀和
定义两个数组:一个a[N],一个b[N]。
我们定义b[N] = a[1]+a[2] + …+a[i]+…+a[N],则b[i]表示a数组的前i项的和,那么b数组为a数组元素的前缀和数组。
对于前缀和,我们有
b[r] = a[1] + a[2]+......+a[r];
b[r] - b[l-1] = a[l]+a[l+1]+......+a[r];//此为数组l到r的数组元素的和
差分
差分可以看成前缀和的逆运算。
我们同样定义两个数组:一个a[N],一个b[N]。
我们定义使得b[N] = a[1]+a[2] + …+a[i]+…+a[N],则我们称a数组为b数组的差分数组。
那么如何构造差分数组呢?
如下所示:
b[0] = 0
a[1] = b[1] - b[0]
a[2] = b[2] - b[1]
......
a[N] = b[N] - b[N-1]
那么我们来说说要用差分解决的问题。
我们给定区间[L,R],我们要使区间[L,R]内的所有元素全部加上x,一般我们用暴力解决这种问题就会使用一重循环,时间复杂度为O(n),如果我们要进行m次这样的操作,时间都复杂度就为O(n*m),这样在我们做题的时候很容易超时。
而我们用差分就可以进行更高效的运算。
我们始终要记得b数组为a数组的前缀和数组,如果我们对a[i]进行修改,必然会对b[i]及其之后的每一个数组元素都会产生影响。
比如:
我们要使得b数组在区间[L,R]内的值都加上x,
那么我们可以使得a[L] + x,则b数组就会有b[L]+x,b[L+1]+x… b[N]+x。
我们再使得a[R+1]-x,这样就有b[R+1]-x,b[R+2]-x…b[N]-x,这一步是为了防止影响b[R]以后的数组元素。
如下图所示:
(我们定义的数组从下标1开始赋值)
比如我们使得b数组内元素为1 6 5 4 3
,那么a数组为1 5 -1 -1 -1 -3
(我们可得b为a的前缀和数组,差分数组要比原数组多一个元素,差分属猪最后一个元素相当于0减去原函数最后一个元素)。
我们要使b数组从下标2到4的元素都加上5,则我们用上述所说的方法来让a[2]+5,让a[5]-5,最后我们再用前缀和方法,进行b[i] = a[i]+a[i-1],最后得到b数组为1 11 10 9 3
达到了让b数组再区间[2,4]内都加上5的目的。这样我们的时间复杂度为O(1)。
总结
要使一位数组在区间[L,R]内的元素都加上x,则我们让他的差分数组c[L]+x,c[R+1]-x。
例如杭电oj里的一道例题Color the ball就是一道差分的运用题。
代码
#include <bits/stdc++.h>
#include <iostream>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 1e6 + 5;
int num[N];
int main()
{
int n;
cin >> n;
while (n != 0)
{
memset(num, 0, sizeof(num));
for (int i = 0; i < n; i++)
{
int l, r;
cin >> l >> r;
num[l]++;
num[r + 1]--;
}
for (int i = 1; i <= n; i++)
{
num[i] += num[i - 1];
cout << num[i];
if(i < n)cout<<" ";
}
cout << endl;
cin >> n;
}
return 0;
}
还有部落外的数也是一道差分的运用题。