动态区间最小值查询,支持点设置,线段树

从周二来镇江出差到现在已经3天了。这几天镇江细雨绵绵,天气异常阴冷,更兼近来事情多有不顺,心情比较沮丧。今天与客户讨论结束,下午闲暇,却心乱如麻,半天时间在焦躁中浪费了。晚上静下来,整理了一下前一阵看的线段树算法:

// 用线段树实现动态区间最小值查询,对源数组A[0..N-1]支持更新操作和查询操作:
// update(int i, int v) 将数组第i个元素值设置为v, i属于区间[0,N)
// query(int L, int R ) 查询数组区间[L,R]的最小值
public class RangeTree4RMQ {
	
	//N为源数组总长度,可查询区间为[0,N)
	int N = 0;		
	
	//M为最底层叶子节点数目,M = min { x | x = 2^k && x >= N }
	int M = 0;
	
	//线段树的数组表示
	int[] tree = null;
	
	public RangeTree4RMQ(int[] A) {
		build(A);
	}
	
	//构造线段树,从叶子节点递推,复杂度O(N)
	public void build(int[] A) {
		
		if ( A == null || A.length <= 0 )
			throw new IllegalArgumentException();
		
		N = A.length;
		M = calculate(N);
		
		tree = new int[2*M-1];
		
		//初始化所有叶子结点
		for ( int i=0; i<N; i++ ) tree[i+M-1] = A[i];
		for ( int i=N; i<M; i++ ) tree[i+M-1] = 0;
		
		//初始化非叶子结点
		for ( int i=M-2; i>=0; i-- ) {
			int x = tree[2*i+1];
			int y = tree[2*i+2];
			tree[i] = (x<y)?x:y;
		}
		
	}
	
	// 区间最小值查询
	// L : 待查询区间左边界
	// R : 待查询区间右边界
	public int query(int L, int R) {
		if ( L > R || L <0 || L >=N || R < 0 || R >=N )
			throw new IllegalArgumentException();
		return query(L, R, 0, 0, M-1);
	}
	
	//区间最小值查询
	//cur : 当前结点在线段树中的编号
	//CUR_L : 当前结点区间的左边界
	//CUR_R : 当前结点区间的右边界
	private int query(int L, int R, int cur, int CUR_L, int CUR_R) {
		
		//待查询区间完全覆盖当前结点区间
		if ( L <= CUR_L && CUR_R <= R ) return tree[cur];
		
		int CUR_M = ( CUR_L + CUR_R ) / 2;
		int res = Integer.MAX_VALUE;
		
		//查询区间与左半区间有重叠
		if ( L <= CUR_M ) {
			int x = query(L, R, 2*cur+1, CUR_L, CUR_M);
			res = (res<x)?res:x;
		}
		
		//查询区间与右半区间重叠
		if ( R > CUR_M ) {
			int x = query(L, R, 2*cur+2, CUR_M+1, CUR_R);
			res = (res<x)?res:x;
		}
		
		return res;
	}
	
	//更新一个值
	public void update(int i, int v) {
		if ( i<0 || i>=N ) 
			throw new IllegalArgumentException();
		update(i, v, 0, 0, M-1);
	}
	
	//更新区间最小值
	//cur : 当前结点在线段树中的编号
	//L : 当前结点区间的左边界
	//R : 当前结点区间的右边界
	private void update(int i, int v, int cur, int L, int R) {
		if ( L == R ) {
			tree[cur] = v;
		} else {
			int M = (L+R)/2;
			if ( i <= M )
				update(i, v, 2*cur+1, L, M);
			else
				update(i, v, 2*cur+2, M+1, R);
			tree[cur] = min(tree[2*cur+1],tree[2*cur+2]);
		}
	}
	
	//取源数组的值A[i]
	public int get(int i) {
		return tree[i+M-1];
	}
	
	// 计算较小值
	private int min(int x, int y) {
		return (x<y) ? x:y;
	}
	
	// 计算最底层的叶子结点数目
	private int calculate(int n) {
		int y = 1;
		while ( y < n ) {
			y <<= 1;
		}
		return y;
	}
	
	// 测试
	public static void main(String[] args) {
		int[] v = { 7, 8, 9, 5, 6, 4, 3, 2, 1};
		RangeTree4RMQ inst = new RangeTree4RMQ(v);
		for ( int i=0; i<v.length; i++ ) {
			System.out.printf("%d ", inst.query(0, i));
		}
		System.out.println();
		for ( int i=0; i<v.length; i++ ) {
			System.out.printf("%d ", inst.query(i, v.length-1));
		}
		System.out.println();
		for ( int i=0; i<v.length; i++ ) {
			inst.update(i, v[i]+1);
			System.out.printf("%d ", inst.query(0, i));
		}
	}

}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值