线段树的修改

对于一棵 最大线段树, 每个节点包含一个额外的 max 属性,用于存储该节点所代表区间的最大值。

设计一个 modify 的方法,接受三个参数 root、 index 和 value。该方法将 root 为根的线段树中 [startend] = [indexindex] 的节点修改为了新的 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);
       
    } 
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值