Segment Tree Build II(線分樹の構造II)

1203 ワード

http://www.lintcode.com/en/problem/segment-tree-build-ii/ Segment Tree Modify(線分樹の修正)を参照してください。最後にmaxの割り当てに注意してください。
/**
 * 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 A: a list of integer
     * @return: The root of Segment Tree
     */
    public SegmentTreeNode build(int[] A) {
        // write your code here
        return tree(A, 0, A.length - 1);
    }

    private SegmentTreeNode tree(int[] a, int start, int end) {
        if (start>end){
            return null;
        }
        SegmentTreeNode node = new SegmentTreeNode(start, end, a[start]);
        if (start == end) {
            return node;
        }
        int mid = (start + end) >>> 1;
        node.left = tree(a, start, mid);
        node.right = tree(a, mid + 1, end);
        
        node.max = Math.max(node.left.max, node.right.max);
        return node;
    }
}