线段树--动态区间查询

线段树是一种数据结构,用于高效地处理数组的区间查询和修改操作。文章介绍了线段树的基本概念,包括建树、修改、查询和返回区间和的功能。线段树的每个节点包含区间左端点、右端点和区间和,通过递归方式初始化和更新。相比于前缀和,线段树更适合动态查询和修改,提供在线算法解决方案。

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

前言:

现在手上有一个数组a[n],让我们来求一个区间的和
这简直太简单了!
我们写一个前缀和就可以在O(1)解决问题了!

问题升级:
有两种操作,一个是修改其中某个变量的值,
另一个是求区间和

这。。我只能O(n²)去做了==

别怕, 线段树可以解决你的问题。

线段树–Segment Tree

既然是线段树,那么一定具有一个树的结构。
这个树和其他树不一样,线段树是用来维护我们手中的数组的

树的结点有这么几个属性:l,r,w
l代表left,是这个结点对应区间的左端点
r代表right,是这个结点对应区间的右端点
w代表这个区间内的区间和

struct node{
    int l;
    int r;
    int w;
}tree[maxn];

那我们来想一下大致的过程吧:
首先我们要对于初始的数组a[n]来把这个线段树建立起来
然后,如果要修改一个值,我们就要修改对应叶子结点和它所有的父节点
查找区间和emmm
把一个大区间分成许多我们维护的小区间就好了~

看上去挺简单的

建树–build_tree()

建树也好,建堆也好,这种操作我们经常使用,就是初始化嘛~
我们从根节点开始初始化,把区间逐渐二分下去。

void build_tree(int x,int left,int right){//x是当前节点标号
    tree[x].l=left;
    tree[x].r=right;
    if(left==right){
        change(x,a[left]);//修改x的值为a[left]
        return ;
    }
    build_tree(2*x,left,(left+right)/2);
    build_tree(2*x+1,(left+right)/2+1,right);
    //左右分别递归调用
    return ;
}
int main(){
    /**/
    build_tree(1,1,n);
    /**/
} 

这样我们就能递归的建立一个线段树了。

修改–change()

刚刚的代码里出现了什么奇怪的东西!
是的,我们调用了还没写的change函数
change,就是把一个节点的值进行修改(现在我们简单的定义为加法)
获取的参数是结点的序号,还有改动的值
然后递归向上修改所有的父亲节点:

void change(int x,int a){//x节点的值修改为tree[x].w+a
    if(x==0) return ;//代表更新完毕
    tree[x].w+=a;
    change(x/2,a);
    return ;
}

查询–find()

那么,我们拿到的指令是,修改a[i]
怎么通过a[i]找到对应的那个叶子结点呢?
二分嘛:

int find(int x,int p,int left,int right){//返回a[x]对应的线段树中的p
    if(left==right) return p;//找到了
    p=(x<=(left+right)/2):p*2?p*2+1;//x在左边,就找左儿子,在右边就找右儿子
    return find(x,p,tree[p].l,tree[p].r);//接着找
}

返回区间和–add()

我们已经可以修改并且维护这颗大树了!
接下来只要算算区间和究竟是多少就好了

我们拿到一个区间,这个区间很大可能没有直接与之对应的节点
往往可能是5个,6个节点拼凑出要求的区间

写几个判断语句,然后递归:

int add(int x,int left,int right){//返回区间和
    if((tree[x].l==left)&&(tree[x].r==right))//正好相等,返回w
        return tree[x].w;
    int mid=(tree[x].l+tree[x].r)/2;
    if(right<=mid) //只可能存在于左儿子
        return add(x*2,left,right);
    if(left>mid) //只可能存在于右儿子
        return add(x*2+1,left,right);
    return add(x*2,left,mid)+add(x*2+1,mid+1,right);//最普遍情况,再次一分为二
}

调用的时候,x为1,因为要从根节点开始寻找

这样我们就可以进行修改和在线查询了。
前缀和求区间和的方法我们称为离线算法。
而今天的线段树是在线算法。
不同的情况,对于在线离线的要求是不一样的。

当然,能写前缀和我绝对不写线段树


最后祝各位OIer武运昌隆!!!

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值