剑指offer 专项突破版 76、数组中的第 k 大的数字

最新推荐文章于 2023-06-17 00:17:39 发布

抓抓璐璐小可爱(๑• . •๑)

最新推荐文章于 2023-06-17 00:17:39 发布

阅读量236

点赞数

分类专栏：算法文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_45883188/article/details/124160727

版权

算法专栏收录该内容

145 篇文章 5 订阅

订阅专栏

本文介绍了一种使用小根堆和快速排序策略解决数组中第k大元素问题的编程方法。通过小根堆实现时间复杂度为O(nlogk)，同时展示了如何巧妙地运用快速排序partition函数。通过实例代码解析了这两种方法在解决该问题上的应用。

摘要由CSDN通过智能技术生成

题目链接

思路

这道题可以用一个大小为k的小根堆来做，这样时间复杂度为O(nlogk)

但是也可以用快速排序~ 一定要注意partition函数的写法哦

  public class Solution {

      private void swap(int[] nums, int left, int right) {
          int tmp = nums[left];
          nums[left] = nums[right];
          nums[right] = tmp;
      }

      private int partition(int[] nums, int left, int right) {
          if (left < right) {
              int random = new Random().nextInt(right - left + 1) + left;
              swap(nums, random, right);

              int pre = left - 1, cur = left;
              while (cur <= right) {
                  if (nums[cur] <= nums[right]) {
                      pre++;
                      swap(nums, pre, cur);
                  }
                  cur++;
              }
              return pre;
          }
          return left;
      }

      public int findKthLargest(int[] nums, int k) {

          int left = 0, right = nums.length - 1, index = partition(nums, 0, nums.length - 1);

          while (index != nums.length - k) {
              if (index < nums.length - k)
                  left = index + 1;
              else
                  right = index - 1;
              index = partition(nums, left, right);
          }
          return nums[index];
      }
  }