【51nod】【树状数组】小陶的疑惑2

最新推荐文章于 2022-01-13 10:19:07 发布

原创最新推荐文章于 2022-01-13 10:19:07 发布 · 152 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树状数组

树状数组专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一道使用树状数组解决的问题——51nod的小陶的疑惑2。文章通过差分数组与树状数组结合的方法，实现了对区间操作的支持，并提供了完整的C++代码实现。

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

【51nod】【树状数组】小陶的疑惑2

题目

在这里插入图片描述

解题思路

差分是通过前缀和来表示当前的数
树状数组可以查询前缀和
问题解决(^ o ^)/

代码

#include<iostream>
#include<cstdio>
using namespace std;
long long n,m,w,x,l,r,b[200010],a[200010];
long long tree[200010];
void add(int y)
{
	 for (;y<=n;y+=y&(-y))
	     tree[y]+=x;
}
long long sum(int y)
{
	long long ans=0;
	for (;y;y-=y&(-y))
	    ans+=tree[y];
	return ans;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	a[1]=0;
	for (int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		b[i]=a[i]-a[i-1];  //求差分数组
		x=b[i];
		add(i);
	}
	for (int i=1;i<=m;i++)
	{
		scanf("%lld",&w);
		if (w==1)
		{
			scanf("%lld%lld%lld",&l,&r,&x);
			add(l);
			x=-x;
			add(r+1);  //差分修改
		}
		else 
		{
		    scanf("%lld",&x);
		    printf("%lld\n",sum(x));
		}
	}
	return 0;
}