前缀和与差分（一维）

前缀和与差分（一维）

前缀和

前缀和是指某些序列的前n项和，在解决某些问题时采用前缀和的思想可以大大的降低时间复杂度

问题引入：输入一个长度为n的数组，元素为a[1] …a[n]，之后进行m次询问，每次询问给出两个值l，r求数组a[l] + a[l+1] +…+a[r]的值。

普通思路：将数组记录下来，每次询问都遍历一遍数组，找到指定的区间，累加输出。这样做的时间复杂度为O（m * n），当数据量过大时，很容易超时，采用前缀和的思想解决这种问题时间复杂度仅为O（n）

算法思想：在查询之前对数组a做一个预处理，创建一个sum数组，sum数组用来记录第i项的前i项和，最后询问某个区间时直接输出a[r] - a[l - 1]即可。核心代码是sum[i] = sum[i - 1] + a[i],每次求前i项和时只需令前一项的和加上a[i]即可

原理：

sum[r] = a[1] + a[2] + a[3] + a[l-1] + a[l] + a[l + 1] … a[r];
sum[l - 1] = a[1] + a[2] + a[3] + a[l - 1];
sum[r] - sum[l - 1] = a[l] + a[l + 1] + …+ a[r];

代码：

#include<iostream>

using namespace std;
int sum[100005] = {0}, a[100005] = {0};
int main()
{
	int n, m;
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
	  cin >> a[i];
	for(int i = 1; i <= n; i++)
	{
		sum[i] = sum[i - 1] + a[i];
	}
	for(int i = 1; i <= m; i++)
	{
		int l, r;
		cin >> l >> r;
		cout << sum[r] - sum[l - 1] << endl;
	}
	return 0;
} 

差分

差分可以看成是前缀和的逆运算，适用于给出一个数组进行多次操作后，输出操作后的数组类的问题

问题引入：给出n个数，将对这n个数进行k次操作，每次操作输入一个区间l和r，对这个区间内的数加上v，进行k次操作后输出这n个数。

思路：第一种暴力的方法我们可以采用每次输入区间都对数组进行修改，最后输出，这样做法的时间复杂度为O（n*k），当数据量过大时有可能超时。我们采用差分的方法进行优化。差分的核心思想如下：

• [l,r] + v == d[l] + v, d[r + 1] - v

• 把标记后的差分数组进行了一次前缀和

  我们用一个数组d来表示arr数组的差分数组，数组d只记录两端的变量，进行k次操作后，将数组d进行前缀和，数组d中就记录了每个数的变量，然后将每个数加它的变量输出即可。优化后的时间复杂度为O（m+n），适用于最后询问次数少的情况。

原理：就是我在前缀和数组的第l位上加v，那么它之后的每个数都会加v，而我在第r+1位上又减去v，那么后面的都会抵消掉，这样sum数组就完成了标记每个数的增量的作用。

代码：

#include<iostream>
#include<bits/stdc++.h>
using namespace std;
int arr[10005], d[10005] = {0};

void add(int l, int r, int v)
{
	d[l] += v;
	d[r + l] -= v;
}

int main()
{
	int n, k;
	cin >> n >> k;
	for(int i = 0; i < n; i++)
	  cin >> arr[i];
	for(int i = 1; i <= k; i++)
	{
		int l, r, v;
		cin >> l >> r >> v;
		add(l, r, v);
	}
	for(int i = 1; i < n; i++)
	  d[i] += d[i - 1];
	for(int i = 0; i < n; i++)
	{
		arr[i] += d[i];
		cout << arr[i] <<" ";
	}
	return 0;
}