1、基本思想
首先定义数组a[] , 其中a[1],a[2]…a[n]作为前缀和。
然后通过a数组来构造数组b,b[1],b[2]…b[n]为差分数组。使得可以通过差分数组的前缀和来表示a数组,即a[i] = b[1] + b[2]+…+b[i]。
构造方式为b[1] = a[1], b[2] = a[2] - a[1], b[n] = a[n] - a[n-1];
得到的关系是:b数组就称为a数组的差分,a数组称为b的前缀和。
2、好处
差分数组的好处是可以简化运算,例如想要给一个区间 [l,r] 上的数组加一个常数c,原始的方法是依次加上c,这样的时间复杂度是O(n)的。但是如果采用差分数组的话,可以大大降低时间复杂度到O(1)。
3、用处
想要在[l,r]区间内的数据同时加一个常数c, 由于a[n] = b[1] + b[2]+…+b[n],因此只需要将b[l] = b[l] + c 即可,这样l之后的数字会依次加上常数c,而在 b[r]处,将b[r+1] = b[r+1] - c ,这样r之后的数组又会恢复原值,仅需要处理这两个边界的差分数组即可,时间复杂度大大降低。
4、代码
假定a数组全部都为0,那么差分数组b也就全为0,
第一次是让原数组中[1,1]区间加上a[1]
第二次是让原数组中[2,2]区间加上a[2]
总结出来 差分就只有一个操作:
如果想给区间[l,r] +c
通用的操作b[l] +c b[r+1] -c
eg:
void insert( int l , int r , int c )
{
b[l] += c;
b[r + 1] -= c;
}
int main()
{
for (int i = 1; i <= n; i++) insert(i,i,a[i]);
}
在for循环中,i = 1,b[1] = a[1] b[2] = -a[1]
i = 2 , b[2] = a[2] - a[1] b[3] = a[3] - a[2]
…
目的就是为了推导出b[1] = a[1], b[2] = a[2] - a[1], b[n] = a[n] - a[n-1];
例题:差分
输入一个长度为 n 的整数序列。
接下来输入 m 个操作,每个操作包含三个整数 l,r,c,表示将序列中 [l,r] 之间的每个数加上 c。
请你输出进行完所有操作后的序列。
输入格式
第一行包含两个整数 n 和 m。
第二行包含 n 个整数,表示整数序列。
接下来 m 行,每行包含三个整数 l,r,c表示一个操作。
输出格式
共一行,包含 n 个整数,表示最终序列。
数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000
输入样例:
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例:
3 4 5 3 4 2
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];
void insert( int l , int r , int c )
{
b[l] += c;
b[r + 1] -= c;
}
int main()
{
scanf("%d%d", &n, &m);
// 输入数组
for (int i = 1; i <= n; i++) scanf("%d",&a[i]);
//初始化b数组
for (int i = 1; i <= n; i++) insert(i,i,a[i]);
while (m --)
{
int l, r, c;
scanf("%d%d%d",&l,&r,&c);
//表示将序列中 [l,r] 之间的每个数加上 c
insert(l,r, c);
}
//b数组的前缀和就是a数组
for (int i = 1; i <= n; i++) b[i] = b[i - 1] + b[i];
for (int i = 1; i <= n; i++) printf( "%d",b[i]);
return 0;
}