五刷215

Author: diguage

docs/0215-kth-largest-element-in-an-array.adoc

@@ -74,6 +74,15 @@ include::{sourcedir}/_0215_KthLargestElementInAnArray_3.java[tag=answer]
 include::{sourcedir}/_0215_KthLargestElementInAnArray_4.java[tag=answer]
 ----
 --
+
+五刷::
++
+--
+[{java_src_attr}]
+----
+include::{sourcedir}/_0215_KthLargestElementInAnArray_5.java[tag=answer]
+----
+--
 ====
 
 == 参考资料

logbook/202503.adoc

@@ -115,6 +115,11 @@ endif::[]
 |{doc_base_url}/0206-reverse-linked-list.adoc[题解]
 |✅ 两种解法：递归+迭代
 
+|{counter:codes2503}
+|{leetcode_base_url}/kth-largest-element-in-an-array/[215. 数组中的第K个最大元素^]
+|{doc_base_url}/0215-kth-largest-element-in-an-array.adoc[题解]
+|✅ 快速选择
+
 |===
 
 截止目前，本轮练习一共完成 {codes2503} 道题。

src/main/java/com/diguage/algo/leetcode/_0215_KthLargestElementInAnArray_4.java

@@ -2,44 +2,44 @@
 
 public class _0215_KthLargestElementInAnArray_4 {
   // tag::answer[]
-/**
- * @author D瓜哥 · https://www.diguage.com
- * @since 2025-03-26 16:24:19
- */
-public int findKthLargest(int[] nums, int k) {
-  // 第 k 大的元素，正好是升序数组的 nums.length - k 个元素
-  return quickselect(nums, nums.length - k, 0, nums.length - 1);
-}
-
-private int quickselect(int[] nums, int k, int left, int right) {
-  if (left == right) {
-    return nums[k];
+  /**
+   * @author D瓜哥 · https://www.diguage.com
+   * @since 2025-03-26 16:24:19
+   */
+  public int findKthLargest(int[] nums, int k) {
+    // 第 k 大的元素，正好是升序数组的 nums.length - k 个元素
+    return quickselect(nums, nums.length - k, 0, nums.length - 1);
   }
-  int pivot = nums[left];
-  // 由于下面先移动指针，所以左右指针各项左右移一位
-  int l = left - 1, r = right + 1;
-  while (l < r) {
-    // 交换之后，没有移动指针，所以先移动指针再循环
-    do {
-      l++;
-    } while (nums[l] < pivot);
-    do {
-      r--;
-    } while (pivot < nums[r]);
-    if (l < r) {
-      int tmp = nums[l];
-      nums[l] = nums[r];
-      nums[r] = tmp;
+
+  private int quickselect(int[] nums, int k, int left, int right) {
+    if (left == right) {
+      return nums[k];
+    }
+    int pivot = nums[left];
+    // 由于下面先移动指针，所以左右指针各项左右移一位
+    int l = left - 1, r = right + 1;
+    while (l < r) {
+      // 交换之后，没有移动指针，所以先移动指针再循环
+      do {
+        l++;
+      } while (nums[l] < pivot);
+      do {
+        r--;
+      } while (pivot < nums[r]);
+      if (l < r) {
+        int tmp = nums[l];
+        nums[l] = nums[r];
+        nums[r] = tmp;
+      }
+    }
+    // r 是最后一个小于或等于 pivot 的元素的索引。
+    // [left, r] 是确定小于或等于 pivot 的部分。
+    if (k <= r) {
+      return quickselect(nums, k, left, r);
+    } else {
+      return quickselect(nums, k, r + 1, right);
     }
   }
-  // r 是最后一个小于或等于 pivot 的元素的索引。
-  // [left, r] 是确定小于或等于 pivot 的部分。
-  if (k <= r) {
-    return quickselect(nums, k, left, r);
-  } else {
-    return quickselect(nums, k, r + 1, right);
-  }
-}
   // end::answer[]
 
   public static void main(String[] args) {

src/main/java/com/diguage/algo/leetcode/_0215_KthLargestElementInAnArray_5.java

@@ -0,0 +1,40 @@
+package com.diguage.algo.leetcode;
+
+public class _0215_KthLargestElementInAnArray_5 {
+  // tag::answer[]
+
+  /**
+   * @author D瓜哥 · https://www.diguage.com
+   * @since 2025-04-02 15:49:20
+   */
+  public int findKthLargest(int[] nums, int k) {
+    return quickSelect(nums, nums.length - k, 0, nums.length - 1);
+  }
+
+  private int quickSelect(int[] nums, int k, int left, int right) {
+    if (left >= right) {
+      return nums[k];
+    }
+    int pivot = nums[left];
+    int l = left - 1, r = right + 1;
+    while (l < r) {
+      do {
+        l++;
+      } while (nums[l] < pivot);
+      do {
+        r--;
+      } while (pivot < nums[r]);
+      if (l < r) {
+        int tmp = nums[l];
+        nums[l] = nums[r];
+        nums[r] = tmp;
+      }
+    }
+    if (k <= r) {
+      return quickSelect(nums, k, left, r);
+    } else {
+      return quickSelect(nums, k, r + 1, right);
+    }
+  }
+  // end::answer[]
+}