0
点赞
收藏
分享

微信扫一扫

树状数组模板

非凡兔 2022-01-06 阅读 62
算法

区间修改区间查询

#include<bits/stdc++.h>
using namespace std;
#define int long long

const int N=1e6+10;
const int M=3e4+3e3;
int n,t,m;
char c;
int sum1[N],sum2[N];

int lowbit(int x)
{
	return x&(-x);
}

void update(int sum[],int id,int x)
{
	for(int i=id;i<=n;i+=lowbit(i))
	sum[i]+=x;
}

int getx(int sum[],int x)
{
	int k=0;
	for(int i=x;i;i-=lowbit(i))
	k+=sum[i];
	return k;
}

signed main()
{
	int x,y,z,s;
	ios::sync_with_stdio(0);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y>>z>>s;
		if(x==1)
		{
			update(sum1,y,s);
			update(sum2,z,s);
		}
		else
		{
			cout<<getx(sum1,z)-getx(sum2,y-1)<<endl;
		}
	}
}

单点修改区间查询

#include<bits/stdc++.h>
using namespace std;

const int N=1e6+10;
const int M=3e4+3e3;
int n,t,m;
char c;
int sum[N];

int lowbit(int x)
{
	return x&(-x);
}

void update(int id,int x)
{
	for(int i=id;i<=n+1;i+=lowbit(i))
	sum[i]+=x;
}

int getx(int x)
{
	int k=0;
	for(int i=x;i;i-=lowbit(i))
	k+=sum[i];
	return k;
}

signed main()
{
	int x,y;
	ios::sync_with_stdio(0);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>c;
		if(c=='B')//单点增加 
		{
			cin>>x>>y;
			update(x,y);
		}
		else if(c=='A')//求0-x的和
		{
			cin>>x;
			cout<<getx(x)<<endl;
		}
		else
		{
			cin>>x>>y;
			update(x,-y);
		}
	}
}```


举报

相关推荐

0 条评论