Leetcode 215.数组中的第k个最大元素

数组中的第k个最大元素

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2

输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4

输出: 4

说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

 class Solution {
     public int findKthLargest(int[] nums, int k) {
         int begin=0;
         int end=nums.length-1;
         k=nums.length+1-k;
         while(begin<end){
             int pos=partition(nums,begin,end);
             if(pos==k-1) break;
             else if(pos<k-1) begin=pos+1;
             else end=pos-1;
         }
         return nums[k-1];
     }
     public int partition(int[] nums, int low, int high) {//分区
         int key;
         if (low > high) {
             return 0;
         }
         key = nums[low]; // 用子表的第一个记录做基准
         while (low < high) { // 从表的两端交替向中间扫描
             while (low < high && nums[high] >= key)
                 high--;
             if (low < high)
                 swap(nums,low,high);// 用比基准小的记录替换低位记录
             while (low < high && nums[low] < key)
                 low++;
             if (low < high) // 用比基准大的记录替换高位记录
                 swap(nums,low,high);
         }
         return low;
     }

     private void swap(int[] a, int i, int j) {
         int t = a[i];
         a[i] = a[j];
         a[j] = t;
     }
 }