区间树(求给定区间是否与已知区间集合相交)

        在x数轴上给出众多线段(已知线段起点与终点),求给定一个线段,并查询该线段与上述众多线段中几条线段相交。创建有序二叉树,节点为众多线段,顺序由线段的起点决定,并且每个节点增加一个域存放该节点的子树中能到达的最右端的值。

class RangeTree {
	private int begin;   //起点
	private int end;	 //终点
	private int maxEnd;  //能到达的最右端

	private RangeTree l;
	private RangeTree r;

	public RangeTree(int begin, int end) {
		this.begin = begin;
		this.end = end;
		this.maxEnd = end;  
	}
	//添加
	public void add(RangeTree x) {
		if (x.begin < begin) {
			if (l == null) {
				l = x;
			} else {
				l.add(x);
			}
		} else {
			if (r == null) {
				r = x;
			} else {
				r.add(x);
			}
		}
		if (l != null && l.end > maxEnd)
			maxEnd = l.end;
		if (r != null && r.end > maxEnd)
			maxEnd = r.end;
	}
	//获取线段相交的数量
	public int getOverCnt(int a, int b) {
		int n = 0;
		if(a >= maxEnd) {
			return 0;
		}
		if (isOver(begin, end, a, b))
			n++;
		if(l!=null) {
			n += l.getOverCnt(a, b);
		}
		if(r!=null) {
			n += r.getOverCnt(a, b);
		}
		return n;
	}
	//判断是否相交
	private boolean isOver(int a1, int b1, int a2, int b2) {
		int len1 = (b1 - a1) + (b2 - a2);
		int maxB = Math.max(b1, b2);
		int minA = Math.min(a1, a2);
		int len2 = maxB - minA;
		return len2 < len1;
	}
}

public class RangeTreeTest {

	public static void main(String[] args) {
		RangeTree root = new RangeTree(3, 5);
		root.add(new RangeTree(1, 3));
		root.add(new RangeTree(2, 4));
		root.add(new RangeTree(5, 7));
		root.add(new RangeTree(8, 10));
		root.add(new RangeTree(4, 8));
		root.add(new RangeTree(2, 5));
		
		int cnt = root.getOverCnt(1, 5);
		System.out.println(cnt);
	}

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值