前缀和+差分的简单用法

本来今天只打算写差分的，但是感觉学了前缀和之后再学差分会容易一点，希望本小白今天的讲解能让你轻松get到它的用法哈~~

还是用我习惯的用例题引入讲解法

（一）前缀和

例题

现在给你一个长度为n的数组，进行m次操作，每次操作都要求出[l,r]区间内所有数的和。

输入:第一行先输入n和m分别表示数组大小以及操作次数

        第二行，输入数组内的每一个数

        接下来m行输入l、和r表示区间

输出：分别l到r区间内的和

一般解法:(就是简单的暴力解法）

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e6+5;
ll a[N];
int main()
{
	ll n,m;
	cin>>n>>m;
	for(ll i=0;i<n;i++){
		cin>>a[i];
	}
	ll l,r;
	for(ll i=1;i<=m;i++){
		cin>>l>>r;
		ll sum=0;
		for(ll j=l-1;j<=r-1;j++){
			sum+=a[j];
		}
		cout<<sum<<endl;
    }
    return 0;
}

这样时间复杂度就为O(n*m),如果n和m的值有很大的话，就会很容易超时，所以如果我们用前缀和来做，时间复杂度即为O(n+m).

前缀和用法：我们可以先定义一个数组sum[i],代表数组a[i]中前i个数相加，那么a[l]到a[r]的和就直接可以用sum[r]-sum[l-1]表示,来看看代码吧

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e6+5;
ll a[N];
ll sum[N];
int main()
{
	ll n,m;
	cin>>n>>m;
	for(ll i=1;i<=n;i++){
		cin>>a[i];
		sum[i]=sum[i-1]+a[i];//注意！不能写成sum[i]+=a[i]的形式 
	}
	ll l,r;
	for(ll i=1;i<=m;i++){
		cin>>l>>r;
		ll b=sum[r]-sum[l-1];
		cout<<b<<endl;//注意换行(本人吃过太多没换行带来的苦了~） 
    }
    return 0;
}

基本上前缀和就是这样用的了，有问题可留言哈~

---

（二）差分

例题：

有N个数，你可以对它进行M次操作，每一次操作给L到R的每个数加K

首先暴力做法就是弄一个for循环在L到R区间内，再则又是执行M次，时间复杂度即为O(N*M），也可能会超时，我们再来看看差分会怎么做。

差分用法：如果有一个数组a:a[1],a[2],a[3]......a[n],此时我们再定义一个数组b:b[1],b[2],b[3]......b[n] 

，使得a[i]=b[1]+b[2]+......b[i],这样数组a即为数组b的前缀和数组，数组b即为数组a的差分数组，我们看看具体是怎么构造的吧

a[0]=0,b[0]=0

a[1]=b[1]                                   所以b[1]=a[1]-a[0]

a[2]=b[1]+b[2]又因为b[1]=a[1] 所以b[2]=a[2]-a[1]

同理b[3]=a[3]-a[2]

.......

b[n]=a[n]-a[n-1]

看了上面的表格，相信大家对差分也有点了解了，我们始终要记住数组a是数组b的前缀和数组，若果我们对b[L]加了一个K，那么从a[L]的每一个数都会加一个K值，但题目要求是我们只需要加到a[R]，如果这样，a[R]以后的每个数也都会加一个数字K，那怎么办呢？我们想想，既然a[R]后面的数多加了一个K，那我们再把它们再减一个K不就OK了吗！即我们把b[R+1]减一个K

总结：若要给数组a中[L.R]每个数加k，只需要对差分数组b进行b[L]+=K,b[R+1]-=K,在来看看代码吧。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll n=1e6+5;
ll a[n],b[n];
int main()
{
	ll N,M;
	cin>>N>>M;
	for(ll i=1;i<=N;i++){
		cin>>a[i];
		b[i]=a[i]-a[i-1];
	}
	ll L,R,K;
	while(M!=0){
		cin>>L>>R>>K;
		b[L]+=K;
		b[R+1]-=K;
		M--;
	}
	for(ll i=1;i<=N;i++){
		b[i]=b[i]+b[i-1];
		cout<<b[i]<<" ";
	}
	return 0;
}

如有问题，欢迎指正哈~~