【洛谷P3368】【模板】树状数组2【树状数组】

stoorz1023

于 2018-08-22 09:16:05 发布

阅读量281

点赞数

CC 4.0 BY-SA版权

分类专栏：树状数组洛谷模板题文章标签：洛谷模板题树状数组

本文链接：https://blog.csdn.net/SSL_ZYC/article/details/81937908

洛谷模板题同时被 2 个专栏收录

22 篇文章

订阅专栏

树状数组

6 篇文章

订阅专栏

本文介绍了一种利用树状数组解决区间加法更新和单点查询问题的方法。通过巧妙地应用前缀和思想，实现了高效的时间复杂度。文章提供了一个具体的洛谷P3368题目的解决方案，并给出了完整的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：

题目链接：https://www.luogu.org/problemnew/show/P3368
已知一个数列，你需要进行下面两种操作：

将某区间每一个数数加上 $x$
输出某一个数

思路：

树状数组只支持单点修改，对于区间修改，单点查询应该如和完成呢？
可以考虑前缀和。每次将一个区间 $[l, r]$ 加上 $x$ 时，就将 $a [l] + = x$ ， $a [r + 1] - = x$ ，这样如果将 $a$ 数组取前缀和，得到的就还是将区间 $[l, r]$ 加 $x$ 。
那么就可以用最基本的树状数组来求解了。

代码：

#include <cstdio>
#define ll long long
using namespace std;

int n,m,w,x,y;
ll a[500001],c[500001],z;

void add(int x,ll y)  //修改
{
	for (;x<=n;x+=x&-x) c[x]+=y;
}

ll ask(int x)  //查询
{
	ll 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]);
		add(i,a[i]-a[i-1]);  //保持前缀和为a数组
	} 
	while (m--)
	{
		scanf("%d",&w);
		if (w==1)  
		{
			scanf("%d%d%d",&x,&y,&z);
			add(x,z);
			add(y+1,-z);  //前缀和
		}
		else
		{
			scanf("%d",&x);
			printf("%d\n",ask(x));
		}
	}
	return 0;
}