树状数组和线段树

EG1:动态求连续区间和

给定 n 个数组成的一个数列,规定有两种操作,一是修改某个元素,二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m,分别表示数的个数和操作次数。

第二行包含 n 个整数,表示完整数列。

接下来 m 行,每行包含三个整数 k,a,b (k=0,表示求子数列[a,b]的和;k=1,表示第 a 个数加 b)。

数列从 1 开始计数。

输出格式
输出若干行数字,表示 k=0 时,对应的子数列 [a,b] 的连续和。

数据范围
1≤n≤100000,
1≤m≤100000,
1≤a≤b≤n
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例:
11
30
35

###法一:树状数组:

#include<bits/stdc++.h>

using namespace std;

const int maxn=100010;

int n,m;
int a[maxn],tr[maxn];//就是c 

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

void add(int x,int v){
    for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=v;
}

int query(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    for(int i=1;i<=n;++i) add(i,a[i]);
    while(m--){
        int k,x,y;
        scanf("%d%d%d",&k,&x,&y);
        if(k==0) printf("%d\n",query(y)-query(x-1));
        else{
            add(x,y);
        }
    }
    return 0;
 } 

###法二:线段树

#include<bits/stdc++.h>

using namespace std;

const int maxn=100010;
int n,m;
int a[maxn];//记录权重
struct node{
	int l,r;//左右区间
	int sum;//总和
}tr[4*maxn];//开 4 倍空间

void pushup(int u){
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;//左儿子 u<<1 ,右儿子 u<<1|1  
}

void build(int u,int l,int r){/*第一个参数为当前节点编号,第二个参数为左边界,第三个参数为右边界*/
	if(l==r) tr[u]={l,r,a[r]};//如果当前已经是叶节点了,那就直接赋值就可以了
	else{//否则的话,说明当前区间长度至少是 2 ,那么需要把当前区间分为左右两个区间,那先要找边界点
		tr[u]={l,r};//记得赋值一下左右边界的初值
		int mid=l+r>>1;//边界直接去计算一下 l + r 的下取整
		build(u<<1,l,mid);//递归一下左儿子
		build(u<<1|1,mid+1,r);//递归一下右儿子
		pushup(u);//更新当前节点信息
	}
}

int query(int u,int l,int r){//从根结点开始往下找对应的一个区间
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;/如果当前区间已经完全被包含了,那么直接返回它的值
    //否则的话需要递归来算
	int mid=tr[u].l+tr[u].r>>1;//计算当前区间中点
    //先判断一下和左边是否有交集
	int sum=0;//用 sum 来表示总和
	if(l<=mid) sum=query(u<<1,l,r);//看一下当前区间的中点和左边有没有交集
	if(r>mid) sum+=query(u<<1|1,l,r);//看一下当前区间的中点和右边有没有交集
	return sum;
}

void modify(int u,int x,int v){//第一个参数是当前节点的编号,第二个参数是要修改的位置,第三个参数是增加的值
	if(tr[u].l==tr[u].r) tr[u].sum+=v;//如果当前已经是叶节点了,那直接让它的总和加上 v 
	else{
		int mid=tr[u].l+tr[u].r>>1;
		 //看一下 x 是在左半边还是在右半边
		if(x<=mid) modify(u<<1,x,v);//如果是在左半边,找左儿子
		else modify(u<<1|1,x,v);//如果在右半边,找右儿子
		//更新完之后当前节点的信息发生变化,需要 pushup 一遍
		pushup(u);
	}
}

int main()
{
	//pushup:用于节点信息更新更新当前节点信息
	//build:在一段区间上初始化线段树
	//modify:单点修改 
	//query:查询
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) scanf("%d",&a[i]); 
	build(1,1,n);/*第一个参数是根节点的下标,根节点是一号点,然后初始区间是 1 到 n */

	while(m--){
		int k,x,y;
		scanf("%d%d%d",&k,&x,&y);
		if(k==0) printf("%d\n",query(1,x,y));//第一个参数是根节点的编号 ,第二个和第三个是我们查询的区间 
        //第一个参数也就是当前节点的编号
		else modify(1,x,y);//第一个参数是根节点的下标,第二个参数是要修改的位置,第三个参数是要修改的值
	}
	return 0;
}

根据评测机数据,树状数组的运行时间仅仅到线段树的一半,且树状数组的代码实现更加简洁。
但线段树功能可以更加强大。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值