首页 > 学院 > 开发设计 > 正文

segment tree(线段树)

2019-11-08 18:24:59
字体:
来源:转载
供稿:网友

简介 
        线段树(segment tree)是一种存储区间的树形结构,方便查询哪一个区间包含了指定的点,原则上,它是一种固定结构,
一旦建成该树,它的结构不会改变。

对于n个区间的集合,建一棵线段树的时间复杂度和空间复杂度都是O(nlgn)。

2. 定义

一维线段树结构:假设有一个区间集合,每一个区间元素从左到右依次是:(-/infty ,p_{1}),[p_{1},p_{1}],(p_{1},p_{2}),[p_{2},p_{2}],...,(p_{m-1},p_{m}),[p_{m},p_{m}],(p_{m},+/infty )。步骤:step1: 将所有的区间元素根据起始点和结束点进行排序;step2: 建一个平衡的二叉搜索树T,height(T)=O(lgn);step3 : 把这些区间插入T中,每插入一个元素需要O(lgn)时间;其中,最左边的叶子节点对应最左边的区间,叶子节点v对应区间Int[v]。

3. 实现

下面是线段树应用的一个例子,segment的区间对应数组的下标表示,值对应区间范围内所有数的总和。
class NumArray {    public:    class segmentTreeNode{        public:        int start,end;        int sum;        segmentTreeNode* left,*right;        segmentTreeNode(int start,int end)        {            this->start=start;            this->end=end;            sum=0;            left=right=NULL;        }    };    segmentTreeNode* build(vector<int>& nums,int s,int e)    {        if(s>e) return NULL;        segmentTreeNode *root=new segmentTreeNode(s,e);        if(s==e) root->sum=nums[s];        else{            int mid=s+(e-s)/2;            root->left=build(nums,s,mid);            root->right=build(nums,mid+1,e);            root->sum=root->left->sum+root->right->sum;        }        return root;    }    void update(segmentTreeNode* root,int i,int val)    {        if(root->start==root->end) root->sum=val;        else        {            int mid=root->start+(root->end-root->start)/2;            if(i<=mid)            update(root->left,i,val);            else update(root->right,i,val);            root->sum=root->left->sum+root->right->sum;        }    }    int sumRange(segmentTreeNode*root,int s,int e)    {        if(root->start==s&&root->end==e)        return root->sum;        int mid=root->start+(root->end-root->start)/2;        if(e<=mid) return sumRange(root->left,s,e);        else if(mid<s) return sumRange(root->right,s,e);        else return sumRange(root->left,s,mid)+sumRange(root->right,mid+1,e);    }    NumArray(vector<int> nums) {        root=build(nums,0,nums.size()-1);    }        void update(int i, int val) {        update(root,i,val);    }        int sumRange(int i, int j) {        int a= sumRange(root,i,j);        cout<<a<<endl;        return a;    }    PRivate:    segmentTreeNode* root;};
转载出于:1.点击打开链接2.点击打开链接


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表