P3368 【模板】树状数组 2

https://www.luogu.com.cn/problem/P3368

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数数加上 xx；
2. 求出某一个数的值。

输入格式

第一行包含两个整数 NN、MM，分别表示该数列数字的个数和操作的总个数。

第二行包含 NN 个用空格分隔的整数，其中第 ii 个数字表示数列第 ii 项的初始值。

接下来 MM 行每行包含 22 或 44个整数，表示一个操作，具体如下：

操作 11： 格式：1 x y k 含义：将区间 [x,y][x,y] 内每个数加上 kk；

操作 22： 格式：2 x 含义：输出第 xx 个数的值。

输出格式

输出包含若干行整数，即为所有操作 22 的结果。

输入输出样例

输入 #1复制

5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4

输出 #1复制

6
10

说明/提示

样例 1 解释：

故输出结果为 6、10。

---

数据规模与约定

对于 30\%30% 的数据：N\le8,M\le10N≤8,M≤10；

对于 70\%70% 的数据：N\le 10000,M\le10000N≤10000,M≤10000；

对于 100\%100% 的数据：1 \leq N, M\le 5000001≤N,M≤500000，1 \leq x, y \leq n1≤x,y≤n，保证任意时刻序列中任意元素的绝对值都不大于 2^{30}230。

---

思路：维护一个差分数组，对差分数组动态前缀和求出在原来数组已经进行了多少操作，然后结合a[x]来输出

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
#define lowbit(x) x&(-x)
using namespace std;
const int maxn=5e5+10;
typedef long long LL;
LL cnt[maxn];
LL a[maxn];
LL n,m;
LL add(LL x,LL d)
{
 	while(x<=n)
 	{
 		cnt[x]+=d;
 		x+=lowbit(x);
	}
} 
LL ask(LL x)
{
	LL sum=0;
	while(x>0)
	{
		sum+=cnt[x];
		x-=lowbit(x);
	}
	return sum;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  cin>>n>>m;
  for(LL i=1;i<=n;i++)
  {
  	cin>>a[i];
  }
  while(m--)
  {
  	LL op,x,y,k;
  	cin>>op;
  	if(op==1)
  	{
  		cin>>x>>y>>k;
  		add(x,k);add(y+1,-k);
	}
	else
	{
		cin>>x;
		cout<<a[x]+ask(x)<<endl;
	}
  }
return 0;
}