#差分
举个栗子,给定一个长度为n的数列,要求针对给定的m个[l,r]区间进行操作,使得该区间内的所有数加X(减X同理),最后给出一个区间[L,R],求出区间内所有元素的和。
对于这个题,暴力是显而易见的,根据题意进行暴力操作就可以,时间复杂度为O(m*n),普通水题还好,一到赛场上这种算法基本要被卡死。
那么,我们需要找到时间复杂度低一点的算法——差分。
差分
对于一个数列a[1],a[2],a[3]…a[n],我们用一个新的数列b去记录a中每两项的差值,即b[1]=a[1]-a[0](a[0]=0),b[2]=a[2]-a[1],b[3]=a[3]-a[2],…,那么就有a[i]=b[1]+b[2]+…+b[i],假定一个位置left,如果 b[left]+=2,那么从 a[left]到a[n],他们都会有a[i]+=2,我们只需要在b[right+1]-=2,就实现了在给定区间[left,right],所有数加2的操作。
代码
const int M=2e5+5;
const int INF=0x3f3f3f3f;
int a[M],b[M],sum[M];
int main()
{
IOS;
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
b[i]=a[i]-a[i-1];//存一个差分数组
}
int m;
cin>>m;
for(int i=1;i<=m;i++)
{
int l,r;
cin>>l>>r;
b[l]+=2;
b[r+1]-=2;
}
int add=0;
for(int i=1;i<=n;i++)
{
add+=b[i];//此时add==a[i]
sum[i]=sum[i-1]+add;//利用前缀和求a[n]的前n项和
}
int L,R;
cin>>L>>R;
cout<<sum[R]-sum[L-1]<<endl;
}
还有一种做法
const int M=2e5+5;
const int INF=0x3f3f3f3f;
int a[M],b[M],sum[M];
int main()
{
IOS;
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
//此处我们不用b[i]存a[i]-a[i-1],
}
int m;
cin>>m;
memset(b,0,sizeof b);
for(int i=1;i<=m;i++)
{
int l,r;
cin>>l>>r;
b[l]+=2;
b[r+1]-=2;
}
int add=0;
for(int i=1;i<=n;i++)
{
add+=b[i];
sum[i]=sum[i-1]+a[i]+add;//前缀和,加上操作后的影响,便是下一项的前缀和
}
int L,R;
cin>>L>>R;
cout<<sum[R]-sum[L-1]<<endl;
}