数据结构:线段树

49 篇文章 0 订阅
8 篇文章 0 订阅

(1)构建



线段树是一棵二叉树,他的每个节点包含了两个额外的属性startend用于表示该节点所代表的区间。start和end都是整数,并按照如下的方式赋值:

  • 根节点的 start 和 end 由 build 方法所给出。
  • 对于节点 A 的左儿子,有 start=A.left, end=(A.left + A.right) / 2
  • 对于节点 A 的右儿子,有 start=(A.left + A.right) / 2 + 1, end=A.right
  • 如果 start 等于 end, 那么该节点是叶子节点,不再有左右儿子。

实现一个 build 方法,接受 start 和 end 作为参数, 然后构造一个代表区间 [start, end] 的线段树,返回这棵线段树的根。


/**
 * Definition of SegmentTreeNode:
 * public class SegmentTreeNode {
 *     public int start, end;
 *     public SegmentTreeNode left, right;
 *     public SegmentTreeNode(int start, int end) {
 *         this.start = start, this.end = end;
 *         this.left = this.right = null;
 *     }
 * }
 */


public class Solution {
    /*
     * @param start: start value.
     * @param end: end value.
     * @return: The root of Segment Tree.
     */
    public SegmentTreeNode build(int start, int end) {
        SegmentTreeNode node = null;
        if(start==end){
            node = new SegmentTreeNode(start, end);
        }
        if(start<end){
            node = new SegmentTreeNode(start, end);
            
            int m = (start+end)/2;
            SegmentTreeNode left = build(start, m);
            SegmentTreeNode right = build(m+1, end);
            
            node.left = left;
            node.right = right;
        }
        
        return node;
    }
}

(2)查询区间最大值



对于一个有n个数的整数数组,在对应的线段树中, 根节点所代表的区间为0-n-1, 每个节点有一个额外的属性max,值为该节点所代表的数组区间start到end内的最大值。

为SegmentTree设计一个 query 的方法,接受3个参数rootstartend,线段树root所代表的数组中子区间[start, end]内的最大值。


package com.graph;

import java.util.*;

import javax.security.auth.kerberos.KerberosKey;

import java.util.*;
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 {
    public int query(SegmentTreeNode root, int start, int end) {

        if(root.start==start && root.end==end){//刚好在该区间内
            return root.max;
        }
        
        int m = (root.start+root.end)/2;
        int leftMax=Integer.MIN_VALUE;
        int rightMax=Integer.MIN_VALUE;
        if(end<=m){//在左子树
            leftMax = query(root.left, start, end);
        }else if(m+1<=start){//在右子树
            rightMax = query(root.right,start, end);
        }else{//横跨左右子树
            leftMax = query(root.left, start, m);
            rightMax = query(root.right, m+1, end);
        }
        
        return Math.max(leftMax, rightMax);
    }
    
    int[] maxnums = new int[] {3,2,2,1,3,0,3};
    int maxIndex=0;
    public SegmentTreeNode build(int start, int end) {
        SegmentTreeNode node = null;
        if(start==end){
            node = new SegmentTreeNode(start, end, maxnums[maxIndex++]);
        }
        if(start<end){
            node = new SegmentTreeNode(start, end, maxnums[maxIndex++]);
            
            int m = (start+end)/2;
            SegmentTreeNode left = build(start, m);
            SegmentTreeNode right = build(m+1, end);
            
            node.left = left;
            node.right = right;
        }
        
        return node;
    }       
    
    public static void main(String[] args) {
		Solution s = new Solution();
		SegmentTreeNode root = s.build(0,3,0);
		s.query(root, 0, 2);
	}
}

(3)修改线段树



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

设计一个 modify 的方法,接受三个参数 root、 index 和 value。该方法将 root 为跟的线段树中 [startend] = [indexindex] 的节点修改为了新的 value ,并确保在修改后,线段树的每个节点的 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 root: The root of segment tree.
     * @param index: index.
     * @param value: value
     * @return: 
     */
    public void modify(SegmentTreeNode root, int index, int value) {
        update(root, index, index, value);    
    }
    
    boolean update(SegmentTreeNode node, int s, int e, int value){
        if(node.start==s && node.end==e){
            node.max=value;
            return true;
        }
        
        boolean flag1=false, flag2=false;
        int m = (node.start+node.end)/2;
        if(e<=m){
            flag1 = update(node.left, s, e, value);
        }else if(m+1<=s){
            flag1 = update(node.right, s, e, value);
        }else{
            flag1 = update(node.left,  s,   m, value);
            flag2 = update(node.right, m+1, e, value);
        }
        
        if(flag1 || flag2){
            int tmpMax = node.max;
            node.max = Math.max(node.left.max, node.right.max);
            
            if(tmpMax!=node.max)
                return true;//node的值更新了
            else 
                return false;
        }else{
            return false;
        }
    }
    
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对算法有兴趣的可以来看看 在自然数,且所有的数不大于30000的范围内讨论一个问题:现在已知n条线段,把端点依次输入告诉你,然后有m个询问,每个询问输入一个点,要求这个点在多少条线段上出现过; 最基本的解法当然就是读一个点,就把所有线段比一下,看看在不在线段中; 每次询问都要把n条线段查一次,那么m次询问,就要运算m*n次,复杂度就是O(m*n) 这道题m和n都是30000,那么计算量达到了10^9;而计算机1秒的计算量大约是10^8的数量级,所以这种方法无论怎么优化都是超时 因为n条线段是固定的,所以某种程度上说每次都把n条线段查一遍有大量的重复和浪费; 线段树就是可以解决这类问题的数据结构 举例说明:已知线段[2,5] [4,6] [0,7];求点2,4,7分别出现了多少次 在[0,7]区间上建立一棵满二叉树:(为了和已知线段区别,用【】表示线段树中的线段) 【0,7】 / \ 【0,3】 【4,7】 / \ / \ 【0,1】 【2,3】 【4,5】 【6,7】 / \ / \ / \ / \ 【0,0】 【1,1】 【2,2】 【3,3】 【4,4】 【5,5】 【6,6】 【7,7】 每个节点用结构体: struct line { int left,right; // 左端点、右端点 int n; // 记录这条线段出现了多少次,默认为0 }a[16]; 和堆类似,满二叉树的性质决定a[i]的左儿子是a[2*i]、右儿子是a[2*i+1]; 然后对于已知的线段依次进行插入操作: 从树根开始调用递归函数insert // 要插入的线段的左端点和右端点、以及当前线段树中的某条线段 void insert(int s,int t,int step)

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值