欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

线段树详解

程序员文章站 2022-07-13 11:19:26
...

很用心写的这篇博客,反正是花了好几天。线段树详解

前言:

要在自然数,且所有的数不大于30000的范围内讨论一个问题:现在已知n条线段,把端点依次输入告诉你,然后有m个(多次)询问,每个询问输入一个点,要求这个点在多少条线段上出现过;

最暴力的解法当然就是读一个点,就把所有线段比一下,看看在不在线段中;但是每次询问都要把n条线段查一次,那么m次询问,就要运算m*n次,复杂度就是O(m*n)

假如m是30000,那么计算量达到了10^9;而计算机1秒的计算量大约是10^8的数量级,所以这种方法无论怎么优化都是超时

因为n条线段是固定的,所以某种程度上说每次都把n条线段查一遍有大量的重复和浪费;

线段树就是可以解决这类问题的数据结构。

类似于二叉树,有构造,插入,修改的方法。

关键是要设置一个结构体,来存放不同的线段,然后依次往后构造或者修改。其中关键是左儿子的标识是父亲标识×2,右儿子的标识是父亲标识×2+1.

对引入的线段树进行详解:

线段树,其实就是把区间逐次二分得到的一种树状结构,它反映了包括归并排序在内的很多分治算法的问题求解方式。

线段树详解

上图是一棵典型的线段树,它对区间[1,10]进行分割,直到单个点。这棵树的特点是:

1. 每一层都是区间[a,b]的一个划分,记L=b-a

2. 一共有log2L层

3. 给定一个点p,从根到叶子p上的所有区间都包含点p,且其他区间都不包含点p。

4. 给定一个区间[l; r],可以把它分解为不超过2log2L条不相交线段的并。

其中第四点并不是很显然,图8.1显示了[2, 8]的分解方式,红色结点为分解后的区间,浅灰色结点是递归分解过程中经过的结点。为了叙述方便,下面称树中的结点所对应的区间为树中区间。

从第3点和第4点可以得出结论:修改一个点只用修改log2 L个树中区间信息,而统计一个区间只需要累加2log2 L个树中区间的信息,且访问的总结点数是O(log L)的。两个操作都很容易实现。


接下来通过做的一道题进行具体化解释分析,帮助理解:

敌兵布阵点击打开链接

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 89431 Accepted Submission(s): 37669

Problem Description
C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视。
*情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动,而Derek每次询问的段都不一样,所以Tidy不得不每次都一个一个营地的去数,很快就精疲力尽了,Derek对Tidy的计算速度越来越不满:"你个死肥仔,算得这么慢,我炒你鱿鱼!”Tidy想:“你自己来算算看,这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下,Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说:“死肥仔,叫你平时做多点acm题和看多点算法书,现在尝到苦果了吧!”Tidy说:"我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼,这么算他真的会崩溃的,聪明的读者,你能写个程序帮他完成这项工作吗?不过如果你的程序效率不够高的话,Tidy还是会受到Derek的责骂的.

Input
第一行一个整数T,表示有T组数据。
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有4种形式:
(1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);
(3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;
(4)End 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令

Output
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

Sample Input

1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End

Sample Output

Case 1: 6 33 59
/*将该题的关系以线段树的结构来建立,可以对建好的线段树(即二叉树),对它上面的
节点从上到下、从左到右依次标号为1、2、3...M,可以看出,节点1的左儿子和右儿子分
别是2和3,以此类推,结点i的左右儿子扥别是2*i、2*i+1,用位运算可以表示为i<<1和
i<<1|1,所以我们可以用一个数组sum[]来表示这个附加的属性,而下标就是节点的标号。
那么这个数组应该申请多大呢,根据树的理论,如果范围是[0,N-1],则M=2*N+1。
*/
/*
题解:线段树:单点更新,只更新叶子结点,然后把信息用PushUp(int r)这个函数更新上来。
线段树功能:update:单点更新;query:区间求和
lson和rson分别表示节点的左儿子和右儿子
PushUp(int  rt)表示把当前结点的信息更新到父节点
rt表示当前子树的根(root),也就是当前所在的结点。
*/
/*
建立以rt为根节点[l,r]为操作区间的线段树

*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<map>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=55555;
int sum[maxn<<2];//用来存放每个点的人数
using namespace std;
void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)//如果l==r,说明已经是叶子节点了,没有儿子节点了,就显现成熟单个营地,人数就是sum[rt].
    {
        scanf("%d",&sum[rt]);
        return;
    }
    int m=(l+r)>>1;
    build(lson);//构造左儿子树
    build(rson);//构造右儿子树
    PushUp(rt);
}
void update(int p,int add,int l,int r,int rt)
{
    if(r==l)//在根节点为rt,[l,r]为操作区间的线段树里对p处的值更新为add 
    {
        sum[rt]+=add;
        return;
    }
    int m=(l+r)>>1;
    if(p<=m)
        update(p,add,lson);
    else update(p,add,rson);
    PushUp(rt);
}

int query(int L,int R,int l,int r,int rt)
{//在根节点为rt,[l,r]为操作区间的线段树里查询区间[L,R]的值
    if(L<=l&&r<=R)
        return sum[rt];//找到要求的线段区间,返回其值
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)
        ret+=query(L,R,lson);//要查询的线段在该线段左边,查询该线段的左子节点
    if(R>m)
        ret+=query(L,R,rson); //要查询的线段在该线段右边,查询该线段的右子节点
    return ret;//要查询的线段在该线段中间,分段查询,左右节点都查。
}
int main()
{
    int T,n;
    scanf("%d",&T);
    for(int i=1;i<=T;i++)
    {
        printf("Case %d:\n",i);
        scanf("%d",&n);
        build(1,n,1);
        char op[10];
        while(scanf("%s",op))
        {
            if(op[0]=='E')
                break;
            int a,b;
            scanf("%d%d",&a,&b);
            if(op[0]=='Q')
                printf("%d\n",query(a,b,1,n,1));
            else if(op[0]=='S')
                update(a,-b,1,n,1);//增加
            else update(a,b,1,n,1);//减少
        }
    }
    return 0;
}