程序员面试金典——维护x的秩___

来源:互联网 发布:java统一协议才能注册 编辑:程序博客网 时间:2024/05/22 20:27

题目描述

现在我们要读入一串数,同时要求在读入每个数的时候算出它的秩,即在当前数组中小于等于它的数的个数(不包括它自身),请设计一个高效的数据结构和算法来实现这个功能。

给定一个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++;}//ifelse {//右侧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;}


原创粉丝点击