SDU_week5_B - TT's Magic Cat（差分）

题目描述

题意

TT 有一只猫 ，它从 世界地图 选了 n 个城市，用 ai 表示每个城市的资产 。
猫会给出几个操作， 区间 [ l , r ] 的城市资产都加 c 。在q次操作后，输出所有城市的资产。

Input
第一行有两个数 n, q (1<=n,q<=2*10^5) ,n表示城市的个数，q表示操作次数。
第二行包含n个数，分别代表每个城市的资产ai。(-10^6 <=ai<=10^6)
接下来q行，每一行代表一个操作, l ,r ,c
(1≤l≤r≤n,−10^5 ≤c≤ 10^5)(1≤ l ≤ r ≤n,−10^5 ≤ c ≤ 10^5)

output
每个城市最后的资产

题目分析

这道题就是差分经典题。构造差分数组B：
B[1] = A[1]
B[i] = A[i] - A[i-1]
从而将对A的操作转化为对B的操作。
差分特点：
B 数组前缀和⇔A 数组元素值：
即SUM{B[1~i]} = A[i] • A
数组的区间加⇔B 数组的单点修改：
即A[L]~A[R] 均加上 c等价于 B[L] += c, B[R+1] -= c

本题复杂度仅为O（2n+q）=O（n+q），优于暴力的O（qn）。

HIT

第一次提交的时候在test17 WA了，检查发现是数组类型范围不够大，将int改为long long后ac。

代码

#define _CRT_SECURE_NO_WARNINGS
#include<cstdio>
#include<iostream>

using namespace std;

//第一次数字开小了，int全换为ll后通过
long long n, q;//n为数字个数，q为操作次数
long long a[1000009];//每个数字的大小
long long b[1000009];//差分数组
long long l, r;//左端点，右端点
long long c;//每次改变的大小
long long ans;//答案变量

int main()
{
	scanf("%lld%lld", &n, &q);
	for (int i = 1; i <= n; i++)
		scanf("%lld", &a[i]);

	if (n == 1)
	{//只有一个数单独处理
		while (q--)
		{
			scanf("%lld%lld%lld", &l, &r, &c);
			a[1] += c;
		}
		printf("%lld", a[1]);
	}

	else
	{//多个数字
		b[1] = a[1];
		for (int i = 2; i <= n; i++)
			b[i] = a[i] - a[i - 1];

		while (q--)
		{
			scanf("%lld%lld%lld", &l, &r, &c);
			b[l] += c;
			if (r!=n)
			{//若r为右端点则不存在b[r+1]
				b[r + 1] -= c;
			}
		}

		ans = 0;
		for (int i = 1; i <= n; i++)
		{
			ans += b[i];
			if (i == 1)printf("%lld", ans);//防CE
			else printf(" %lld", ans);
			
		}
	}
	return 0;
}