题目描述
现在我们要读入一串数,同时要求在读入每个数的时候算出它的秩,即在当前数组中小于等于它的数的个数(不包括它自身),请设计一个高效的数据结构和算法来实现这个功能。
给定一个int数组A,同时给定它的大小n,请返回一个int数组,元素为每次加入的数的秩。保证数组大小小于等于5000。
测试样例:
[1,2,3,4,5,6,7],7
返回:[0,1,2,3,4,5,6]
思路:构造新的树的数据结构,主要是在树的数据结构中新加了一个左子节点个数。
构造二叉树,从根节点开始,一依次将小于等于 父节点的放左侧,大于放右侧,依次进行,放置到最合适的位置。
如果放在左侧,则该次的父节点的leftsize+1,主义leftsize只存自己左侧子(重点是子)节点个数。右侧不用加。
然后使用getLeftNum从头开始依次往下取到他的leftsize。
import java.util.*;
public class Rank {
public int[] getRankOfNumber(int[] A, int n) {
// write code here
int result[]=new int[n];
if(n==0){result[n-1]=0;return result;}
buildTree(A, n,result);
return result;
}
public void buildTree(int[]A,int n,int[] result){
Node head = new Node(A[0]);
int count = 1;
while(count<n){//1-n-1个子节点
Node newnode = new Node(A[count]);
insertTree(head,newnode);
result[count]=getLeftNum(head,newnode.val);
count++;
}
}
public void insertTree(Node head,Node newnode){
if(newnode.val<=head.val){//判定左侧
if(head.left!=null){
insertTree(head.left, newnode);
}else{
head.left=newnode;
}
head.leftsize++;
}//if
else {//右侧
if(head.right!=null){
insertTree(head.right, newnode);
}else {
head.right = newnode;
}
}//else
}
public int getLeftNum(Node head,int val){
if(val==head.val)return head.leftsize;
else if(val<head.val){
return getLeftNum(head.left, val);
}else{
return head.leftsize+1+getLeftNum(head.right, val);
}
}
}
class Node{
public Node(int val){
this.val = val;
}
int leftsize =0;
int val;
Node left,right;
}