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

lintcode-medium-Segment Tree Query

时间:2016-04-05 19:14:24      阅读:191      评论:0      收藏:0      [点我收藏+]

标签:

For an integer array (index from 0 to n-1, where n is the size of this array), in the corresponding SegmentTree, each node stores an extra attribute max to denote the maximum number in the interval of the array (index from start to end).

Design a query method with three parameters rootstart and end, find the maximum number in the interval [start, end] by the given root of segment tree.

 

Example

For array [1, 4, 2, 3], the corresponding Segment Tree is:

                  [0, 3, max=4]
                 /                       [0,1,max=4]        [2,3,max=3]
          /         \        /            [0,0,max=1] [1,1,max=4] [2,2,max=2], [3,3,max=3]

query(root, 1, 1), return 4

query(root, 1, 2), return 4

query(root, 2, 3), return 3

query(root, 0, 2), return 4

 

/**
 * Definition of SegmentTreeNode:
 * public class SegmentTreeNode {
 *     public int start, end, max;
 *     public SegmentTreeNode left, right;
 *     public SegmentTreeNode(int start, int end, int max) {
 *         this.start = start;
 *         this.end = end;
 *         this.max = max
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    /**
     *@param root, start, end: The root of segment tree and 
     *                         an segment / interval
     *@return: The maximum number in the interval [start, end]
     */
    public int query(SegmentTreeNode root, int start, int end) {
        // write your code here
        
        if(start > end || end < root.start || start > root.end)
            return 0;
        
        if(root.start == start && root.end == end)
            return root.max;
        
        int leftmax = Integer.MIN_VALUE;
        int rightmax = Integer.MIN_VALUE;
        
        int mid = root.start + (root.end - root.start) / 2;
        
        if(start <= mid){
            if(end <= mid){
                leftmax = query(root.left, start, end);
            }
            else{
                leftmax = query(root.left, start, mid);
            }
        }
        
        if(end > mid){
            if(start > mid){
                rightmax = query(root.right, start, end);
            }
            else{
                rightmax = query(root.right, mid + 1, end);
            }
        }
        
        return Math.max(leftmax, rightmax);
    }
}

 

lintcode-medium-Segment Tree Query

标签:

原文地址:http://www.cnblogs.com/goblinengineer/p/5356105.html

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