leetcode215（数组中的第K个最大元素）

最新推荐文章于 2022-03-11 12:01:08 发布

AI算法工程师YC

最新推荐文章于 2022-03-11 12:01:08 发布

阅读量110

点赞数

分类专栏： LeetCode 文章标签： leetcode215 数组中的第K个最大元素

本文链接：https://blog.csdn.net/qq_36134437/article/details/102688884

版权

LeetCode 专栏收录该内容

142 篇文章 2 订阅

订阅专栏

数组中的第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 ≤ 数组的长度。

*/C++解法

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());
        return nums[nums.size()-k];
        
    }
};

*/

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int> q(nums.begin(), nums.end());
        for (int i = 0; i < k -1;++i){
            q.pop();
        }
        return q.top();
    }
};

*/

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size() - 1;
        while (true) {
            int pos = partition(nums, left, right);
            if (pos == k - 1) return nums[pos];
            if (pos > k - 1) right = pos - 1;
            else left = pos + 1;
        }
    }
    int partition(vector<int>& nums, int left, int right) {
        int pivot = nums[left], l = left + 1, r = right;
        while (l <= r) {
            if (nums[l] < pivot && nums[r] > pivot) {
                swap(nums[l++], nums[r--]);
            }
            if (nums[l] >= pivot) ++l;
            if (nums[r] <= pivot) --r;
        }
        swap(nums[left], nums[r]);
        return r;
    }
};

*/Python解法
class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        return self.find(nums, k-1, 0,len(nums)-1)
    def find(self, nums, k ,left ,right):
        l = left
        r = right
        partition = nums[(left+right)//2]
        while(left <= right):
            while(nums[left] > partition):
                left +=1
            while(nums[right] < partition):
                right-=1
            if left <= right:
                nums[left] ,nums[right] = nums[right], nums[left]
                left +=1
                right -=1
        if right >= k and right > l:
            return self.find(nums, k ,l ,right)
        if left <= k and left < r:
            return self.find(nums, k ,left, r)
        return nums[k]
            
*/Python解法
class Solution:
    def findKthLargest(self,nums, k):
        nums.sort()
        return nums[-k]
*/Python 解法
class Solution:
    def findKthLargest(self, nums, k):
        low, high = 0, len(nums)-1
        while low <= high:
            pivot = self.partition(nums, low ,high)
            if pivot == k-1:
                return nums[pivot]
            if pivot < k-1:
                low = pivot + 1
            else:
                high = pivot -1
    def partition(self, nums, low, high):
        pivot_value = nums[high]
        index = low
        for i in range(low ,high):
            if nums[i] >= pivot_value:
                nums[i], nums[index] = nums[index], nums[i]
                index+=1
        nums[index], nums[high] = nums[high], nums[index]
        return index