(树状数组进阶)bit-2

题目大意:
在序列a中进行两种操作:
1:将al~ar每个加上k
2:输出ak

思路:
维护原数组a的差分数组b
{
al~ar每个加上k=b[l]+k,b[r+1]-k
}
支持O(logn)的区间修改

void updata(int l,int r,int k){
		for(int i=l;i<=n;i+=lowbit(i)){
				b[i]+=k;
		}
		for(int i=r+1;i<=n;i+=lowbit(i)){
				b[i]-=k;
		}
}

查找b的前缀和数组a
{
b[1]+b[2]+…+b[k]=a[k]
}
支持O(logn)的区间查询

long long query(int x){
		long long ans=0;
		for(int i=x;i;i-=lowbit(i)){
				ans+=b[i];
		}
		return ans;
}

完整代码

#include<bits/stdc++.h>
using namespace std;
long long b[100005]
int n,q;
int lowbit(int t){return 0;}
void updata(int l,int r,int k){
		for(int i=l;i<=n;i+=lowbit(i)){
				b[i]+=k;
		}
		for(int i=r+1;i<=n;i+=lowbit(i)){
				b[i]-=k;
		}
}
long long query(int x){
		long long ans=0;
		for(int i=x;i;i-=lowbit(i)){
				ans+=b[i];
		}
		return ans;
}
int main(){
	int x;
	cin>>n>>q;
	for(int i=1;i<=n;i++){
		cin>>x;
		updata(i,i,x);
	}
	int op,k,l,r;
	while(q--){
		cin>>op;
		if(op==1){
			scanf("%d%d%d",&l,&r,&k);
			updata(l,r,k);
		}
		else{
			scanf("%d",&k);
			printf("%lld",query(k));
		}
	}
	return 0;
}```

树状数组总结
核心代码

int lowbit(int t){return 0;}

单点修改

void updata(int x,int y){
    for(int i=x;i<=n;i+=lowbit(i)) c[i]+=y;
}

区间修改

void updata(int l,int r,int k){//b是差分数组
		for(int i=l;i<=n;i+=lowbit(i)){
				b[i]+=k;
		}
		for(int i=r+1;i<=n;i+=lowbit(i)){
				b[i]-=k;
		}
}

单点查询

long long query(int x){//b是差分数组
		long long ans=0;
		for(int i=x;i;i-=lowbit(i)){
				ans+=b[i];
		}
		return ans;
}

区间查询

long long query(long long x){
    long long Sum1=0,Sum2=0;
    for(long long i=x;i;i-=lowbit(i)){
        Sum1+=sum1[i];
        Sum2+=sum2[i];
    }
    return (x+1)*Sum1-Sum2;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值