215. Kth Largest Element in an Array

首先复习一下quicksort.

先理解partition函数，就是选序列的最后一个数作为pivot，比它小的排到它左边，比它大的在它右边，它在中间。但是它左右的部分并没有排序。

返回值是这个pivot所在的位置，pivot所在位置就是它最终的位置，也就是说比如index是5，说明它就是第6大的数。

quicksort里面就是调用partition,不断对其左右进行排序，直到low >= high

partition伪代码

paritition(arr, low, hight):

　　index = low;//index是第一个不小于pivot的数的index

　　对于从low到high的数：

　　　　如果这个数比pivot小，那么：
　　　　　　把index和这个数交换
　　　　　　index++
　　swap pivot和index所在的数

返回index

 

 

quicksort伪代码

quicksort(arr, low, high):

　　如果low<high:

　　　　partitionIndex = partition(arr, low, high);
　　　　partition(arr, low, partitionIndex - 1);
　　　　partition(arr, partitionIndex + 1, high);

 

 

所以这道题的思路就是，先partition一次，然后如果正好是第k-1个,那就返回，不然如果partition完左侧加起来的数字数目已经超过了k-1，那么说明右侧反正都小于这个pivot了，就只对左侧递归，不然就对右侧递归

   public int findKthLargest(int[] nums, int k) {
        if (nums.length == 1) {
          return nums[0];
        }
        int idx = -1;
        int left = 0, right = nums.length - 1;
        while (idx != k - 1) {
          idx = partition(nums, left, right);
          if(idx > k - 1) {
               right = idx - 1;
          } else {
              left = idx + 1;
          }
        }
        return nums[idx];
   }
    
    private int partition(int[] nums, int low, int high) {
        int index = low;
        int pivot = nums[high];
        for(int i = low; i < high; i++) {
            if(nums[i] > pivot) {
                swap(nums, index, i);
                index++;
            }
        }
        swap(nums, index, high);
        return index;
    }
    
    private void swap(int[] nums, int idx1, int idx2) {
        int tmp = nums[idx1];
        nums[idx1] = nums[idx2];
        nums[idx2] = tmp;
    }

 

另外一种方法是用PriorityQueue。

PQ的值得记录的：

1.自动维护从小到大的顺序，可以poll(),peek().但是不方便得到某一位置上的元素。

2.当然可以自己写comparator，但是如果单纯想逆序的话，可以new PQ<>(Collections.reverseOrder())

    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
        for(int i = 0; i < nums.length; i++) {
            pq.offer(nums[i]);
            if(pq.size() > k) {
                pq.poll();
            }
        }
        return pq.poll();
    }