141、數據流中的第K大元素

題目描述
設計一個找到數據流中第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。

有技巧，但是沒想到怎麼來，我的代碼

class KthLargest {

    	int d;
        int []x;
    
    public KthLargest(int k, int[] nums) {
        d = k;
        x = new int[nums.length];
        x = nums;
    }
    
    public int add(int val) {
    	x = Arrays.copyOf(x, x.length + 1);
        x[x.length - 1] = val;
        Arrays.sort(x);
        return x[ x.length - d];
    }
}

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */

嘗試進行改進

class KthLargest {

 int result = Integer.MIN_VALUE;    int []x;
    int d = 0;
    public KthLargest(int k, int[] nums) {
        x = nums;
        Arrays.sort(x);
        d = k;
    //    result = x[nums.length - k];
    }
    
    public int add(int val) {
    	if(val < result){
    		return result;
    	}else {
			x = Arrays.copyOf(x, x.length + 1);
			x[x.length - 1] = val;
			Arrays.sort(x);
			result = x[x.length - d];
			return result;
			
		}
    }
}

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */

然鵝並沒有怎麼進行優化
排名靠前的代碼
使用了一個新的api叫做優先級隊列，顧名思義，該隊列使用的是從小到大進行的排序，其含義是如果你一直放入元素，那麼最小的元素會在最上面，即peek是最上面的，因此該代碼中的隊列的長度是k並且第k大元素就是該隊列的最小值，也就是最小值

class KthLargest {
    PriorityQueue<Integer> queue = null;
    int size = 0;

    public KthLargest(int k, int[] nums) {
        queue = new PriorityQueue<Integer>(k);
        size = k;
        
        for (Integer num : nums)
            add(num);
    }
    
    public int add(int val) {
        if (queue.size() < size)
            queue.offer(val);
        else {
            if (queue.peek() < val) {
                queue.poll();
                queue.offer(val);
            }
            
        }
        return queue.peek();
    }
}