week5作业题_B-TT's Magic Cat

B-TT’s Magic Cat

---

题目描述

长度为n的数组，一共q次操作，1 <= n，q <= 1e5。每次操作给出L，R，c表示区间[L，R]中各个数均加上c，求q次操作后，数组中各个元素值。
Input
第一行输入两个正整数n，q表示数组长度和操作的次数。第二行输入n个数字，表示数组中的n个元素ai（-1e6 <= ai <= 1e6）。接下来的q行输入三个数字L，R，c（-1e5 <= c <= 1e5）

题目思路

首先我们可以想到的一个方法是暴力做法，直接循环将每个数都加上c，这样的复杂度是O(qn)，会直接TLE。另外一个我我们可以想到的办法就是前缀和与差分。我们可以将原数组转化为差分数组，将区间修改转变为单点修改，然后再使用前缀和就可以了。
差分数组：
b[0] = a[0];
b[i] = a[i] - a[i - 1];
这样a[i]就等于差分数组b的前i项的和。在差分数组中我们可以将a[L，R] + c转化为b[L] + c，b[R] - c。最后输出前缀和就行。

代码实现

#include <bits/stdc++.h>
using namespace std;
#define _for(i,a,b) for(long long int i = (a);i < (b); i++)
#define _rep(i,a,b) for(long long int i = (a);i <= (b); i++)
#define ll long long
const int N = 2e5 + 10;

ll n,q;
ll a[N],b[N],sum[N];		//原数组，差分数组，前缀和数组 
ll l,r,c;

int main()
{
	cin >> n >> q;
	_rep(i,1,n)
		cin >> a[i];
	b[1] = a[1];
	_rep(i,2,n)
		b[i] = a[i] - a[i-1];
	_rep(i,1,q)
	{
		cin >> l >> r >> c;
		b[l] += c;			//单点修改 
		b[r+1] -= c;
	}
	sum[1] = b[1];
	_rep(i,2,n)
		sum[i] = sum[i-1] + b[i];		//求前缀和 
	_rep(i,1,n)
		cout << sum[i] << " ";
}

---

总结

在这个题目我们需要注意到数据大小的问题，这里为了避免超出int的数据范围，我使用的是long long。