【洛谷P3374 P3368】树状数组

最新推荐文章于 2022-09-28 16:39:26 发布

cout0

最新推荐文章于 2022-09-28 16:39:26 发布

阅读量109

点赞数

分类专栏： C/C++ 文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/u011956367/article/details/119821075

版权

C/C++ 专栏收录该内容

72 篇文章 6 订阅

订阅专栏

这两篇博客介绍了树状数组（也称为二进制指数树）的基础知识和使用技巧。第一篇展示了如何利用树状数组进行区间加法操作和求和查询。第二篇则讲解了差分的概念，并演示了如何在区间修改操作中利用树状数组，通过更新差分数组实现高效计算。

摘要由CSDN通过智能技术生成

提示

本文只记录模板，不做详细解释

P3374 树状数组1

原题链接

#include <iostream>
#define lowbit(x) x&(-x)
#define N 5*(int)1e5+1
using namespace std;
int n,m,num[N];
long long tree[N];
void add(int idx,int val){
	while(idx<=n){
		tree[idx]+=val;
		idx+=lowbit(idx);
	}
}
long long sum(int idx){
	long long ret=0;
	while(idx){
		ret+=tree[idx];
		idx-=lowbit(idx);
	}
	return ret;
}
int main() {
    scanf("%d%d",&n,&m);
    fill(tree,tree+n+1,0);
    for(int i=1;i<=n;i++){
		scanf("%d",&num[i]);
		add(i,num[i]);
	}
	for(int i=1;i<=m;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(x==1){
			add(y,z);
		}else{
			printf("%lld\n",sum(z)-sum(y-1));
		}
	}
    return 0;
}

P3368 树状数组2

原题链接

#include <iostream>
#define lowbit(x) x&(-x)
#define N 5*(int)1e6+1
using namespace std;
int n,m,num[N];
long long tree[N];
void add(int idx,int val){
	while(idx<=n){
		tree[idx]+=val;
		idx+=lowbit(idx);
	}
}
long long sum(int idx){
	long long ret=0;
	while(idx){
		ret+=tree[idx];
		idx-=lowbit(idx);
	}
	return ret;
}
int main() {
    scanf("%d%d",&n,&m);
    fill(tree,tree+n+1,0);
    for(int i=1;i<=n;i++){
		scanf("%d",&num[i]);
		add(i,num[i]-num[i-1]);
	}
	for(int i=1;i<=m;i++){
		int c,l,r,k;
		scanf("%d",&c);
		if(c==1){
			scanf("%d%d%d",&l,&r,&k);
			add(l,k);
			add(r+1,-k);
		}else{
			scanf("%d",&k);
			printf("%lld\n",sum(k));
		}
	}
    return 0;
}
/*
差分
设数组a[]={1,6,8,5,10}，那么差分数组b[]={1,5,2,-3,5}
也就是说b[i]=a[i]-a[i-1];(a[0]=0;)，那么a[i]=b[1]+....+b[i];(这个很好证的)。
假如区间[2,4]都加上2的话
a数组变为a[]={1,8,10,7,10}，b数组变为b={1,7,2,-3,3};
发现了没有，b数组只有b[2]和b[5]变了，因为区间[2,4]是同时加上2的,所以在区间内b[i]-b[i-1]是不变的.
所以对区间[x,y]进行修改,只用修改b[x]与b[y+1]:
b[x]=b[x]+k;b[y+1]=b[y+1]-k;
*/