数组中第K大元素

discriminator

于 2024-03-30 15:10:51 发布

阅读量235

点赞数 5

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/discriminator/article/details/137174093

版权


// class Solution {
// public:
//     int findKthLargest(vector<int>& nums, int k) {
//         int n = nums.size();
//         // 第k个最大元素在排序数组中的位置是n-k
//         return quickSelect(nums, 0, n - 1, n - k);
//     }

//     int quickSelect(vector<int>& nums, int left, int right, int k_smallest) {
//         if (left == right) // 如果数组包含一个元素
//             return nums[left]; // 返回这个元素

//         int pivotIndex = partition(nums, left, right);
        
//         // pivot正好是第k小的元素
//         if (k_smallest == pivotIndex)
//             return nums[k_smallest];
//         // 第k小的元素在pivot左边
//         else if (k_smallest < pivotIndex)
//             return quickSelect(nums, left, pivotIndex - 1, k_smallest);
//         // 第k小的元素在pivot右边
//         return quickSelect(nums, pivotIndex + 1, right, k_smallest);
//     }

// int partition(vector<int>& nums, int left, int right) {
//     // 使用随机数生成器选择枢轴
//     srand(time(NULL)); // 初始化随机数种子
//     int pivotIndex = left + rand() % (right - left + 1); // 随机选择一个索引作为枢轴
//     int pivot = nums[pivotIndex];
//     // 将枢轴移到数组的最左侧
//     swap(nums[pivotIndex], nums[left]);

//     int i = left + 1;

//     for (int j = left + 1; j <= right; j++) {
//         if (nums[j] < pivot) {
//             swap(nums[i], nums[j]);
//             i++;
//         }
//     }
//     // 将枢轴放到它最终的位置
//     swap(nums[left], nums[i - 1]);
//     return i - 1; // 返回枢轴的最终位置
// }

// };


/*
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        // 对数组进行排序，使用默认的升序排序
        sort(nums.begin(), nums.end());
        // 第k个最大元素在排序后的数组中的位置是n-k
        // 因为排序是升序的，所以需要从数组末尾开始计算位置
        return nums[nums.size() - k];
    }
};
*/

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        // 构建最大堆
        make_heap(nums.begin(), nums.end());
        // 移除前k-1个最大元素
        for (int i = 0; i < k - 1; ++i) {
            pop_heap(nums.begin(), nums.end() - i); // 移动当前最大元素至区间末尾
             //nums.pop_back(); // 实际不需要从vector中移除元素
        }
        // 此时最大堆的根节点是第k个最大元素
        return nums.front();
    }
};

// class Solution {
// public:
//     int findKthLargest(vector<int>& nums, int k) {
//         // 直接找到第k大的元素，不需要转换为第n-k小
//         nth_element(nums.begin(), nums.begin() + k - 1, nums.end(), greater<int>());
//         return nums[k - 1];
//     }
// };