码迷,mamicode.com
首页 > 其他好文 > 详细

白话数据结构之【线段树】

时间:2015-02-17 18:48:03      阅读:135      评论:0      收藏:0      [点我收藏+]

标签:

线段树的入门


      线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
      对于线段树中的每一个非叶子节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度。
      使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。

----来自百度百科
【以下以 求区间最大值为例】

先看声明:

#include <cstdio>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 400000;
const int MAX = 1000003;
struct NODE{
    int value;        // 结点对应区间的权值,注意有些题目里面这个值可以有多个,改成数组就行了,比如cf291的D题
    int left,right;   // 区间 [left,right]
}node[maxn];
int father[MAX];     // 每个点(当区间长度为0时,对应一个点)对应的结构体数组下标

技术分享

【创建线段树(初始化)】:

       由于线段树是用二叉树结构储存的,而且是近乎完全二叉树的,所以在这里我使用了数组来代替链表上图中区间上面的红色数字表示了结构体数组中对应的下标。

在完全二叉树中假如一个结点的序号(数组下标)为 I ,那么 (二叉树基本关系)

I 的父亲为 I/2,

I 的另一个兄弟为 I/2*2 或 I/2*2+1

I 的两个孩子为 I*2 (左)   I*2+1(右)

有了这样的关系之后,我们便能很方便的写出创建线段树的代码了。

void BuildTree(int i,int left,int right){ // 为区间[left,right]建立一个以i为祖先的线段树,i为数组下标,我称作结点序号
    node[i].left = left;    // 写入第i个结点中的 左区间
    node[i].right = right;  // 写入第i个结点中的 右区间
    node[i].value = 0;      // 每个区间初始化为 0
    if (left == right){ // 当区间长度为 0 时,结束递归
        father[left] = i; // 能知道某个点对应的序号,为了更新的时候从下往上一直到顶
        return;
    }
    // 该结点往 左孩子的方向 继续建立线段树,线段的划分是二分思想,如果写过二分查找的话这里很容易接受
    // 这里将 区间[left,right] 一分为二了
    int mid = (left + right)/2;
    BuildTree(i<<1, left, mid);
    // 该结点往 右孩子的方向 继续建立线段树
    BuildTree(i<<1|1, mid + 1, right);
}

【单点更新线段树】:

       由于我事先用 father[ ] 数组保存过 每单个结点 对应的下标了,因此我只需要知道第几个点,就能知道这个点在结构体中的位置(即下标)了,这样的话,根据之前已知的基本关系,就只需要直接一路更新上去即可。

void UpdataTree(int ri){ // 从下往上更新(注:这个点本身已经在函数外更新过了)

    if (ri == 1)return; // 向上已经找到了祖先(整个线段树的祖先结点 对应的下标为1)
    int fi = ri / 2;        // ri 的父结点
    int a = node[fi<<1].value; // 该父结点的两个孩子结点(左)
    int b = node[fi<<1|1].value; // 右
    node[fi].value = max(a,b);    // 更新这个父结点(从两个孩子结点中挑个大的)
    UpdataTree(ri/2);       // 递归更新,由父结点往上找
}



【查询区间最大值】:
       将一段区间按照建立的线段树从上往下一直拆开,直到存在有完全重合的区间停止。对照图例建立的树,假如查询区间为 [2,5] 
技术分享

红色的区间为完全重合的区间,因为在这个具体问题中我们只需要比较这 三个区间的值 找出 最大值 即可。

int Max = -1111;
void Query(int i,int l,int r){ // i为区间的序号(对应的区间是最大范围的那个区间,也是第一个图最顶端的区间,一般初始是 1 啦)
    if (node[i].left == l && node[i].right == r){ // 找到了一个完全重合的区间
         Max = max(Max,node[i].value);
         return;
    }
    int ans = 0;
    i = i << 1; // get the left child of the tree node
    if (l <= node[i].right){ // 左区间有涉及
        if (r <= node[i].right) // 全包含于左区间,则查询区间形态不变
            Query(i, l, r);
        else // 半包含于左区间,则查询区间拆分,左端点不变,右端点变为左孩子的右区间端点
            Query(i, l, node[i].right);
    }
    i += 1; // right child of the tree
    if (r >= node[i].left){ // 右区间有涉及
        if (l >= node[i].left) // 全包含于右区间,则查询区间形态不变
            Query(i, l, r);
        else // 半包含于左区间,则查询区间拆分,与上同理
            Query(i, node[i].left, r);
    }
}


下面给一道模板题验证程序的正确性

HDU1754

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 42585    Accepted Submission(s): 16851


Problem Description
很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
 

Input
本题目包含多组测试,请处理到文件结束。
在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取‘Q‘或‘U‘) ,和两个正整数A,B。
当C为‘Q‘的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。
当C为‘U‘的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。
 

Output
对于每一次询问操作,在一行里面输出最高成绩。
 

Sample Input
5 6 1 2 3 4 5 Q 1 5 U 3 6 Q 3 4 Q 4 5 U 2 9 Q 1 5
 

Sample Output
5 6 5 9
Hint
Huge input,the C function scanf() will work better than cin
 
#include <cstdio>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 2097152;
const int MAX = 1000003;
struct NODE{
    int value;        // 结点对应区间的权值
    int left,right;   // 区间 [left,right]
}node[maxn];
int father[MAX];     // 每个点(当区间长度为0时,对应一个点)对应的结构体数组下标
int ID[maxn];


void BuildTree(int i,int left,int right){ // 为区间[left,right]建立一个以i为祖先的线段树,i为数组下标,我称作结点序号
    node[i].left = left;    // 写入第i个结点中的 左区间
    node[i].right = right;  // 写入第i个结点中的 右区间
    node[i].value = 0;      // 每个区间初始化为 0
    if (left == right){ // 当区间长度为 0 时,结束递归
        father[left] = i; // 能知道某个点对应的序号,为了更新的时候从下往上一直到顶
        return;
    }
    // 该结点往 左孩子的方向 继续建立线段树,线段的划分是二分思想,如果写过二分查找的话这里很容易接受
    // 这里将 区间[left,right] 一分为二了
    int mid = (left + right)/2;
    BuildTree(i<<1, left, mid);
    // 该结点往 右孩子的方向 继续建立线段树
    BuildTree(i<<1|1, mid + 1, right);
}


void UpdataTree(int ri){ // 从下往上更新(注:这个点本身已经在函数外更新过了)

    if (ri == 1)return; // 向上已经找到了祖先(整个线段树的祖先结点 对应的下标为1)
    int fi = ri / 2;        // ri 的父结点
    int a = node[fi<<1].value; // 该父结点的两个孩子结点(左)
    int b = node[fi<<1|1].value; // 右
    node[fi].value = max(a,b);    // 更新这个父结点(从两个孩子结点中挑个大的)
    UpdataTree(ri/2);       // 递归更新,由父结点往上找
}

int Max = -1111;
void Query(int i,int l,int r){ // i为区间的序号(对应的区间是最大范围的那个区间,也是第一个图最顶端的区间,一般初始是 1 啦)
    if (node[i].left == l && node[i].right == r){ // 找到了一个完全重合的区间
         Max = max(Max,node[i].value);
         return;
    }
    int ans = 0;
    i = i << 1; // get the left child of the tree node
    if (l <= node[i].right){ // 左区间有涉及
        if (r <= node[i].right) // 全包含于左区间,则查询区间形态不变
            Query(i, l, r);
        else // 半包含于左区间,则查询区间拆分,左端点不变,右端点变为左孩子的右区间端点
            Query(i, l, node[i].right);
    }
    i += 1; // right child of the tree
    if (r >= node[i].left){ // 右区间有涉及
        if (l >= node[i].left) // 全包含于右区间,则查询区间形态不变
            Query(i, l, r);
        else // 半包含于左区间,则查询区间拆分,与上同理
            Query(i, node[i].left, r);
    }
}

int main()
{
    #ifdef xxz
    freopen("in.txt","r",stdin);
    #endif // xxz
    int n,m,grade;
    while(~scanf("%d%d",&n,&m))
    {
        BuildTree(1,1,n);//先建树
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&grade);
            node[father[i]].value = grade;//用father数组来定位叶子节点在哪,且赋值
            UpdataTree(father[i]);//然后更新
        }

        while(m--){
            char ch[3];int a,b;
            scanf("%s %d %d",ch,&a,&b);
            if (ch[0] == 'Q'){
                Max = 0;
                Query(1, a, b);
                printf("%d\n",Max);
            }else{
                node[father[a]].value = b; // 底层的无条件更新成绩
                UpdataTree(father[a]);
            }
        }


    }
    return 0;
}




白话数据结构之【线段树】

标签:

原文地址:http://blog.csdn.net/u013445530/article/details/43866799

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!