《算法竞赛进阶指南》0x42 T2 A Tiny Problem with Integers

题目传送门

题目描述

给定长度为 N 的数列 A，然后输入 M 行操作指令。

第一类指令形如 C l r d，表示把数列中第 l∼r 个数都加 d。

第二类指令形如 Q x，表示询问数列中第 x 个数的值。

对于每个询问，输出一个整数表示答案。

题解

一道区间修改，单点查询的树状数组板子题
此树状数组维护的不再是每个元素的值
由于是区间修改的问题，为了减少时间复杂度，最容易想到的其实就是差分，而此树状数组维护的就是差分序列
在朴素算法中，当我们进行了多次区间修改，再求某点的值时，会在这个点初始值的基础上，加上差分数组的前缀和
所以只需要维护出差分数组的前缀和，就可以进行单点查询操作，而对于差分数组而言，区间修改也只是改变两端点的值而已，因此问题就转化为了针对差分数组进行单点修改，区间查询

code

#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=1e5+10;
int a[N],c[N];
void change(int x,int y)
{
	for(;x<=n;x+=x&-x) c[x]+=y;
}
int ask(int x)
{
	int ans=0;
	for(;x;x-=x&-x) ans+=c[x];
	return ans;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=m;i++)
	{
		char c[3];
		scanf("%s",c);
		if(c[0]=='C')
		{
			int l,r,d;
			scanf("%d%d%d",&l,&r,&d);
			change(l,d);
			change(r+1,-d);
		}
		else 
		{
			int x;
			scanf("%d",&x);
			printf("%d\n",a[x]+ask(x));
		}
	}
	return 0;
}