码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode-215. Kth Largest Element in an Array

时间:2019-04-30 23:35:31      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:tco   return   https   find   output   com   turn   ted   may   

地址:https://leetcode.com/problems/kth-largest-element-in-an-array/

描述:

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

 

实例1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5

实例2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

Note: 
You may assume k is always valid, 1 ≤ k ≤ array‘s length.

 

方法:

public int findKthLargest(int[] nums, int k) {
        int start = 0, end = nums.length - 1, index = nums.length - k;
        while (start < end) {
            int pivot = partion(nums, start, end);
            if (pivot < index) start = pivot + 1;
            else if (pivot > index) end = pivot - 1;
            else return nums[pivot];
        }
        return nums[start];
    }

    private int partion(int[] nums, int start, int end) {
        int pivot = start, temp;
        while (start <= end) {
            while (start <= end && nums[start] <= nums[pivot]) start++;
            while (start <= end && nums[end] > nums[pivot]) end--;
            if (start > end) break;
            temp = nums[start];
            nums[start] = nums[end];
            nums[end] = temp;
        }
        temp = nums[end];
        nums[end] = nums[pivot];
        nums[pivot] = temp;
        return end;
    }

 

Leetcode-215. Kth Largest Element in an Array

标签:tco   return   https   find   output   com   turn   ted   may   

原文地址:https://www.cnblogs.com/zimengfang/p/10798272.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!