Interval Minimum Number

LintCode-205.Interval Minimum Number

Given an integer array (index from 0 to n-1, where n is the size of this array), and an query list. Each query has two integers [start, end]. For each query, calculate the minimum number between index start and end in the given array, return the result list.

Example

For array [1,2,7,8,5], and queries [(1,2),(0,4),(2,4)], return [2,1,5]

Challenge

O(logN) time for each query

/**
 * Definition of Interval:
 * public classs Interval {
 *     int start, end;
 *     Interval(int start, int end) {
 *         this.start = start;
 *         this.end = end;
 *     }
 */
public class Solution {  
    /**
     *@param A, queries: Given an integer array and an query list
     *@return: The result list
     */
    public ArrayList<Integer> intervalMinNumber(int[] A, 
                                                ArrayList<Interval> queries) {
        if (A == null || A.length == 0 
            || queries == null || queries.size() == 0) {
            return new ArrayList<Integer>();        
        }

        SegmentTreeNode root = buildTree(0, A.length - 1, A);

        ArrayList<Integer> result = new ArrayList<Integer>();
        for (Interval interval : queries) {
            int min = query(root, interval.start, interval.end);
            result.add(min);
        }

        return result;
    }

    int query(SegmentTreeNode node, int start, int end) {
        if (node == null) {
            return 0;
        }

        if (start <= node.start && end >= node.end) {
            return node.min;
        }

        int middle = node.start + (node.end - node.start) / 2;

        if (end <= middle) {
            return query(node.left, start, end);
        } else if (start > middle) {
            return query(node.right, start, end);
        } else {
            return Math.min(query(node.left, start, middle), query(node.right, middle + 1, end));
        }
    }

    SegmentTreeNode buildTree(int start, int end, int[] A) {
        if (start > end) {
            return null;
        }

        SegmentTreeNode node = new SegmentTreeNode(start, end);

        if (start != end) {
            int middle = start + (end - start) / 2;

            node.left = buildTree(start, middle, A);
            node.right = buildTree(middle + 1, end, A);

            node.min = Math.min(node.left.min, node.right.min);
        } else {
            node.min = A[start];
        }

        return node;
    }
}

class SegmentTreeNode {  
    int start, end;
    int min;
    SegmentTreeNode left;
    SegmentTreeNode right;

    public SegmentTreeNode(int start, int end) {
        this.start = start;
        this.end = end;
        this.min = 0;
        this.left = this.right = null;
    }
}

Hope this helps,
Michael