程序设计Week5——B-TT‘s magic cat

题目描述

给定一组数，接下来q次操作，每次操作对原数组中下标l到r的所有数加上c，求最后得到的数组。

Input

第一行两个数n，q（1< = n,q< =2e5），表示有n个数，q次操作；
第二行n个数，代表原始给定的数组，-1e6< = ai < =1e6；
接下来q行，每行三个数l、r、c，代表从l到r的数都加上c，1< = l< = r< =n，-1e5 < =c< =1e5。

Output

输出一行n个数，代表q次操作完成后，得到的新数组。

解题思路

普通的遍历加法的方法肯定超时，复杂度为O（nq），使用差分数组，将原本的区间修改简化为单点修改，再利用前缀和，求出修改后的数组值，输出即可。
差分数组构建方式：

• b[1] = a[1]
• b[i] = a[i] - a[i-1]

区间修改变单点修改：

• a[l] ~ a[r] 加上c等价于 b[l] +=c,b[r+1] -=c。

实现代码

#include<iostream>
using namespace std;

long long a[200005],b[200005];

int main()
{
	int n,q;
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;++i)
	{
		scanf("%lld",&a[i]);
		b[i]=a[i]-a[i-1];
	}
	for(int i=0;i<q;++i)
	{
		long long l,r,c;
		scanf("%lld%lld%lld",&l,&r,&c);
		b[l]+=c;
		b[r+1]-=c;
	}
	for(int i=1;i<n;++i)
	{
		a[i]=b[i]+a[i-1];
		printf("%lld ",a[i]);
	}
	printf("%lld\n",b[n]+a[n-1]);
	return 0;
}

总结

本题较为简单，是一个很直白的差分，知道差分数组的构建以及如和修改区间值，套板子即可。