对于一棵 最大线段树, 每个节点包含一个额外的 max
属性,用于存储该节点所代表区间的最大值。
设计一个 modify
的方法,接受三个参数 root
、 index
和 value
。该方法将 root 为根的线段树中 [start, end] = [index, index] 的节点修改为了新的 value ,并确保在修改后,线段树的每个节点的 max 属性仍然具有正确的值。
线段树不清楚的可以自行百度
对于这个修改思路其实很简单,利用递归,找到start等于end并且和index相等的,修改max的值为value,此时递归再次调用,因为修改了max的值可能导致root节点的max变化,所以start不等于end时就要获取left和right的最大值,取最大值,依次递归下去。对于线段树的构造和查询也是如此,都是利用递归。
public class Solution {
/*
* 线段树构造
* @param start: start value.
* @param end: end value.
* @return: The root of Segment Tree.
*/
public SegmentTreeNode build(int start, int end) {
// write your code here
if(start > end){
return null;
}
SegmentTreeNode root = new SegmentTreeNode(start,end);
if(start != end){
int mid = (start + end)/2;
root.left = build(start,mid);
root.right = build(mid+1,end);
}
return root;
}
/**
* 线段树查询
* @param root: The root of segment tree.
* @param start: start value.
* @param end: end value.
* @return: The maximum number in the interval [start, end]
*/
public int query(SegmentTreeNode root, int start, int end) {
if(root==null){
return Integer.MIN_VALUE;
}
if(root.start>end || root.end<start || start >end){
return Integer.MIN_VALUE;
}
if(start<=root.start && end>=root.end){
return root.max;
}
int rootMid = (root.start+root.end)/2;
int leftMax = query(root.left,start,Math.min(rootMid,end));
int rightMax = query(root.right,Math.max(rootMid,start),end);
return Math.max(leftMax,rightMax);
}
/**
* 线段树修改
* @param root: The root of segment tree.
* @param index: index.
* @param value: value
* @return: nothing
*/
public void modify(SegmentTreeNode root, int index, int value) {
if(root==null){
return;
}
if(root.start>index || root.end< index){
return;
}
if(root.start == root.end && root.start == index){
root.max = value;
return;
}
if(root.left.end>=index){
modify(root.left,index,value);
}else if(root.right.start<=index){
modify(root.right, index, value);
}
root.max = Math.max(root.left.max,root.right.max);
}
}