zoukankan      html  css  js  c++  java
  • [LeetCode 215.] 数组中的第K个最大元素

    LeetCode 215. 数组中的第K个最大元素

    题目描述

    给定整数数组 nums 和整数 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

    提示:

    • 1 <= k <= nums.length <= 104
    • -104 <= nums[i] <= 104

    解题思路

    快速查找算法。
    基于快速排序的 partition 算法,每次选定一个 pivot 并调整到合适位置之后,判定其是否为第 k 大元素。不是则进入一半分支内继续查找。

    参考代码

    class Solution {
    public:
        int findKthLargest(vector<int>& nums, int k) {
            if (k <= 0 || k > nums.size()) return -1;
            return findKth(nums, 0, nums.size()-1, k);
        }
        int findKth(vector<int>& nums, int start, int stop, int k) {
            int x = nums[start];
            int i = start, j = stop;
            while (i < j) {
                while (i < j && nums[j] >= x) j--;
                if (i < j) nums[i++] = nums[j];
                while (i < j && nums[i] <= x) i++;
                if (i < j) nums[j--] = nums[i];
            }
            nums[j] = x;
            if (j == nums.size() - k) return nums[j];
            if (j < nums.size() - k) return findKth(nums, j+1, stop, k);
            return findKth(nums, start, j-1, k);
        }
    };
    
  • 相关阅读:
    接口
    java基础
    java的反射
    按照字典序打印所有的字符串
    求幂的问题
    时间复杂度与空间复杂度
    孩子们的游戏(圆圈中最后剩下的数)
    约瑟夫环问题
    翻转单词顺序列
    复杂链表的复制
  • 原文地址:https://www.cnblogs.com/zhcpku/p/15056668.html
Copyright © 2011-2022 走看看