差分数组详解[通俗易懂]

全栈程序员站长

发布于 2022-06-27 20:58:37

5190

发布于 2022-06-27 20:58:37

文章被收录于专栏：全栈程序员必看全栈程序员必看

大家好，又见面了，我是你们的朋友全栈君。

题目：

来先看一道裸题，有n个数。

m个操作,每一次操作，将x~y区间的所有数增加z；

最后有q个询问，每一次询问求出x~y的区间和。

思路：

~~很明显，直接用前缀和无法快速满足这个操作，所以我们就用到了查分数组。~~

设a数组表示原始的数组；

设d[i]=a[i]-a[i-1](1<i≤n,d[1]=a[1])；

设f[i]=f[i-1]+d[i](1<i≤n,f[1]=d[1]=a[1])；

设sum[i]=sum[i-1]+f[i](1<i≤n,sum[1]=f[1]=d[1]=a[1])。

则易知

举个例子，我们求1~3的区间和.

则

后面的可以依次类推。

那么，对于一个操作，我们可以让d[x]加上z，让d[y+1]减小z，就可以了。

还用刚才的例子。

则

后面的可以依次类推。

代码：

#include<cstdio>
	int n,m,q;
	int a[100000],d[100000],f[100000],sum[100000];
int main()
{
	int x,y,z;
	scanf("%d %d %d",&n,&m,&q);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		d[i]=a[i]-a[i-1];
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d %d",&x,&y,&z);
		d[x]+=z;
		d[y+1]-=z;
	}		
	for(int i=1;i<=n;i++)
	{
		f[i]=f[i-1]+d[i];
		sum[i]=sum[i-1]+f[i];
	}
	for(int i=1;i<=q;i++)
	{
		scanf("%d %d",&x,&y);
		printf("%d\n",sum[y]-sum[x-1]);
	}
}

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/133084.html原文链接：https://javaforall.cn

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2022年6月9，如有侵权请联系 cloudcommunity@tencent.com 删除

java