【树状数组】一个简单的整数问题(C++)

题目

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

第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。

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

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

输入格式

第一行包含两个整数N和M。

第二行包含N个整数A[i]。

接下来M行表示M条指令,每条指令的格式如题目描述所示。

输出格式

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

每个答案占一行。

数据范围

1≤N,M≤1051≤N,M≤105,
|d|≤10000|d|≤10000,
|A[i]|≤1000000000|A[i]|≤1000000000

输入样例:

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例:

4
1
2
5




看到这道题后,第一的反应:树状数组。
树状数组只支持单点修改和单点查询,所以在这里,我们要用一个小技巧:
我们可以将C l r d命令改为:
1、在l的位置上加上d
2、在r+1的位置上减去d(即加上-d)
接着,我们发现在l到r的位置上都加上了d,所以这个技巧就是这样。
写完后RE了不知道多少次,还是不知道是什么意思(别的刷题网站上段错误显示为Runtime Error,这个网站上显示为Segmentation Fault)
代码(注释部分为调试部分):

#include<bits/stdc++.h>
#pragma GCC optimize(3)
using namespace std;
long long n;
long long c[100001],a[100001];
void add(long long x,long long y)
{
	for(;x;x-=x&-x)
	c[x]+=y;
}
long long ask(long long x)
{
	long long ans=0;
	for(x;x<=n;x+=x&-x)
	ans+=c[x];
	return ans;
}
int main() {
	long long m;
	scanf("%lld%lld",&n,&m);
	for(long long i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	}
	for(long long i=1;i<=m;i++)
	{
		char s;
		cin>>s;
		if(s=='C')
		{
			long long l,r,d;
			scanf("%lld%lld%lld",&l,&r,&d);
			add(r,d);
			add(l-1,-d);
		}
		if(s=='Q')
		{
			long long d;
			scanf("%lld",&d);
			printf("%lld\n",ask(d)+a[d]);
		}
		/*for(long long j=1;j<=n;j++)
		{
			printf("%lld ",c[j]);
		}
		cout<<endl;
		for(long long j=1;j<=n;j++)
		{
			printf("%lld ",a[j]);
		}
		cout<<endl;*/
	}

    return 0;
}

猜你喜欢

转载自blog.csdn.net/liuzich/article/details/107538124