LeetCode HOT100系列题解之数组中的第K个最大元素（7/100）

Visionmaker

已于 2024-09-08 16:21:03 修改

阅读量977

点赞数 18

分类专栏： LeetCode Hot100 文章标签： leetcode

于 2024-09-08 16:04:02 首次发布

本文链接：https://blog.csdn.net/qq_63747498/article/details/142026731

版权

LeetCode Hot100 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目：第K个最大元素. - 力扣（LeetCode）

题解

方法一快速排序

方法二桶排序

思考：各个排序的思路，以及时间复杂度是多少？

1. 冒泡排序（Bubble Sort）

2. 选择排序（Selection Sort）

3. 插入排序（Insertion Sort）

4. 快速排序（Quick Sort）

5. 归并排序（Merge Sort）

6. 堆排序（Heap Sort）

题目：第K个最大元素. - 力扣（LeetCode）

题解

方法一快速排序

第K大数本身，这是一个经典的快速排序问题，以此做一个快速排序的复习。

需要注意的是，这个题严格规定时间为O(N)，快排则是期望为线性O（N）的。

快速排序的思路：

选取一个基准（pivot），通过一趟排序将数据分成两部分：左边的都小于等于基准，右边的都大于基准。然后递归地对左右两部分进行快速排序。

举例(如果有错，理解就可以）：

6 5 3 4 2 1

l = 0, r = 5

i = -1， j = 6

x = q[0] = 6

while循环结束

i = 0 j = 5

swap(q[0], q[5])

k=4<=5,继续,递归走if

1 5 3 4 2 6

l = 0, r = 5

i = - 1，j = 6

x = q[0] = 1

while(i < j)

while(q [ ++ i ] < 1) 不存在 i = 0

while(q [ - - j ] > 1) j = 0

k = 4 >= 0

走else l = j + 1 = 1, r = 5

1 5 3 4 2 6

l = 1, r = 5

x = q[1] = 5

i = 0, j = 6

while(i < j)

while(q [ ++ i ] < 5) 不存在 i = 1

while(q [ - - j ] > 1) j = 4

swap(q[1], q[4])

1 2 3 4 5 6

l = 1, r = 4

i = 0, j = 5

x = q[1] = 2

i = j = 1

l = 2 ,r = 4

···

l = 4, r = 4 return q[4]

代码：

class Solution {
public:
    int quick_select(vector<int>& q, int l, int r, int k) {
        if (l == r) return q[l];  // 只有一个元素时直接返回

        int i = l - 1, j = r + 1;
        int x = q[l];  // 选择第一个元素作为基准
        while (i < j) {
            while (q[++i] < x);  // 向右找第一个不小于 x 的元素
            while (q[--j] > x);  // 向左找第一个不大于 x 的元素
            if (i < j) swap(q[i], q[j]);  // 交换
        }

        // 判断 k 在左半部分还是右半部分
        if (k <= j) return quick_select(q, l, j, k);  // 在左半部分查找
        else return quick_select(q, j + 1, r, k);  // 在右半部分查找
    }

    int findKthLargest(vector<int>& nums, int k) {
        int n = nums.size();
        return quick_select(nums, 0, n - 1, n - k);  // 查找第 n-k 个最小的元素
    }
};

方法二桶排序

该题给出的数据范围以及时间限制，实际上桶排序更符合要求。

代码：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        vector<int> s(20010, 0);  // 初始化计数数组，大小为20010，用于存储[-10000, 10000]范围内的数字频率

        // 统计每个元素出现的次数
        for (int num : nums) {
            s[num + 10000]++;  // 将原数组元素映射到[0, 20010)的范围
        }

        // 从大到小遍历查找第k大元素
        for (int i = 20009; i >= 0; i--) {  // 从最大值开始遍历
            k -= s[i];
            if (k <= 0) {
                return i - 10000;  // 将索引还原为原数组的数值
            }
        }

        return -1;  // 如果没有找到（理论上不应到达这里）
    }
};