您的位置:首页 > 编程语言 > Java开发

【Java】假设你正在读取一串整数,每隔一段时间,你希望找出数字X的秩(小于或等于x的值的数目)。

2015-07-22 16:37 676 查看
假设你正在读取一串整数,每隔一段时间,你希望找出数字X的秩(小于或等于x的值的数目)。

请事先数据结构和算法支持这些操作。也就是说,实现track(int  x)方法,每读入一个数字都会调用该方法,以及getRankOfNumber(int x),返回元素值小于等于x的元素个数(不包括x本身)

用一个数组存放好所有已排序的元素,getRankOfNumber(int x)实现很简单,执行二分查找,返回索引即可

但当有新元素进来时,需要搬移其他元素腾出空间,插入就会很低效。

我们需要一种数据结构既能维持相对排列顺序还能方便更新元素,那么就是二叉查找树 。

找出某个元素的秩,可以中序遍历,并且在访问结点时用计数器记录好数目。

在查找x时,只要向左移动,计数器变量就不会变,因为右边的都比x大。

但是向右移动时,跳过了左边的一些元素,他们都比x小,这时计数器需要更新。

public class RankNode {
public int left_size = 0;
public RankNode left, right;
public int data = 0;
public RankNode(int d) {
data = d;
}

public void insert(int d) {
if ( d <= data) {
if (left != null) left.insert(d);
else left = new RankNode(d);
left_size++;
}
else {
if (right != null) right.insert(d);
else right = new RankNode(d);
}
}
public int getRank(int d) {
if ( d == data ) {
return left_size;
}
else if ( d < data ) {
if ( left == null ) return -1;
else return left.getRank(d);
else {
int right_rank = right == null ? -1 :right.getRank(d);
if (right_rank == -1) return -1;
else return left_size + 1 + right_rank;
}
}
}
}

public class Question {
private static RankNode root = null;
public static void track(int number) {
if (null == root) {
root = new RankNode(number);
}
else {
root.insert(number);
}
}

public static int getRankOfNumber(int number) {
return root.getRank(number);
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: