lintcode 439. 线段树的构造 II

柔情只为你懂 2023-07-06 09:56 6阅读 0赞

线段树是一棵二叉树，他的每个节点包含了两个额外的属性start和end用于表示该节点所代表的区间。start和end都是整数，并按照如下的方式赋值:

根节点的 start 和 end 由 build 方法所给出。  
对于节点 A 的左儿子，有 start=A.left, end=(A.left + A.right) / 2。  
对于节点 A 的右儿子，有 start=(A.left + A.right) / 2 + 1, end=A.right。  
如果 start 等于 end, 那么该节点是叶子节点，不再有左右儿子。  
对于给定数组实现build方法, 线段树的每个节点储存区间最大值, 返回根节点.

样例
    输入: [3,2,1,4]
    解释: 
    这颗线段树将会是
              [0,3](max=4)
              /          \
           [0,1]         [2,3]    
          (max=3)       (max=4)
          /   \          /    \    
       [0,0]  [1,1]    [2,2]  [3,3]
      (max=3)(max=2)  (max=1)(max=4)
    说明
    线段树(又称区间树)是一种高级数据结构, 支持一系列区间查询/修改操作.
    
    维基百科: https://zh.wikipedia.org/zh-hans/線段樹

/**
     * Definition of SegmentTreeNode:
     * class SegmentTreeNode {
     * public:
     *     int start, end, max;
     *     SegmentTreeNode *left, *right;
     *     SegmentTreeNode(int start, int end, int max) {
     *         this->start = start;
     *         this->end = end;
     *         this->max = max;
     *         this->left = this->right = NULL;
     *     }
     * }
     */
    
    class Solution {
        
    public:
        /**
         * @param A: a list of integer
         * @return: The root of Segment Tree
         */
        SegmentTreeNode * build(vector<int> &A) {
        
            // write your code here
            if(A.size()<=0) return NULL;
            return build(A,0,A.size()-1);
        }
        SegmentTreeNode*build(vector<int>&A,int start,int end)
        {
        
            SegmentTreeNode*node;
            if(start!=end)
                node=new SegmentTreeNode(start,end,0);
            else
                node=new SegmentTreeNode(start,end,A[start]);
            if(start==end) return node;
            node->left=build(A,start,(start+end)/2);
            node->right=build(A,(start+end)/2+1,end);
            node->max=max(node->left->max,node->right->max);
            return node;
        }
    };