【洛谷-P3368】（树状数组）

题目链接：https://www.luogu.org/problemnew/show/P3368#sub

题目描述

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

1.将某区间每一个数数加上x

2.求出某一个数的值

输入输出格式

输入格式：

 

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

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

接下来M行每行包含2或4个整数，表示一个操作，具体如下：

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

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

 

输出格式：

 

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

 

输入输出样例

输入样例#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

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

样例说明：

故输出结果为6、10

思路：

树状数组的区间修改和点查询，主要思路是借助了差分数组，然后将其转换成点更新和区间查询。设d[i]=a[i]-a[i-1]，那么查询一个点x的值就是从d[1]+..+d[x]。d[1]=a[1],d[2]=a[2]-a[1]....。区间修改，对一个区间同时增加一个数，区间内部相邻两点之间差值不变，而左端点和区间外点的值会增加val,右端点差值会减少val。

学习博客：

https://blog.csdn.net/bestsort/article/details/80796531#%E6%A0%91%E7%8A%B6%E6%95%B0%E7%BB%84%E5%9F%BA%E7%A1%80

里面写给a[l]加x，我感觉应该是给d[l]+x。

ac代码：

//洛谷-P3368
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 500100;
int c[maxn],a[maxn];//a-原数组，c-差值数组 
int lowbit(int x){
	return (x&(-x));
}
void update(int x,int val,int n){//点修改 
	for(int i=x;i<=n;i+=lowbit(i)){
		c[i]+=val;
	}
}
void update2(int l,int r,int val,int n){//区间修改注意修改l和r+1 
	update(l,val,n);
	update(r+1,-val,n);
}
int query(int x){//点查询-对差值数组区间查询查询 
	int res=0;
	for(int i=x;i>0;i-=lowbit(i)){
		res+=c[i];
	}
	return res;
}
int main(){
	int n,m;
	scanf("%d%d\n",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		update(i,a[i]-a[i-1],n);
	}
	for(int i=0;i<m;i++){
		int op;
		scanf("%d",&op);
		if(op==1)
		{
			int x,y,k;
			scanf("%d%d%d",&x,&y,&k);
			update2(x,y,k,n);
		}
		else {
			int x;
			scanf("%d",&x);
			printf("%d\n",query(x));
		}
	}
	return 0;
}