问题描述:
设计一个找到数据流中第K大元素的类(class)。注意是排序后的第K大元素,不是第K个不同的元素。
你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器,它包含数据流中的初始元素。每次调用 KthLargest.add,返回当前数据流中第K大的元素。
示例:
int k = 3;
int[] arr = [4,5,8,2];
KthLargest kthLargest = new KthLargest(3, arr);
kthLargest.add(3); // returns 4
kthLargest.add(5); // returns 5
kthLargest.add(10); // returns 5
kthLargest.add(9); // returns 8
kthLargest.add(4); // returns 8
说明:
你可以假设 nums 的长度≥ k-1 且k ≥ 1。
基本思路:
博主一开始遇到这道题的时候还在写BST。所以就照着BST的思路写了。
既然是BST,那就肯定要取思考一个递归的写法。
首先必然是要构造一个BST,碰到一个元素就加入到这个BST中。这个我们之前已经写到过了。
随之而来的就是查找。这里我给BST的节点又新添加了一个属性——以他为根节点的节点总数。
这样以后,比根节点大的元素个数就是它右子树的节点总数,比他小的元素个数就是它左子树的节点总数。
为了进行递归,我们不妨这样做:
- 如果k>右子树的节点总数&#x