【Hot100】LeetCode—215. 数组中的第K个最大元素

目录

  • [1- 思路](#1- 思路)
  • [2- 实现](#2- 实现)
    • [⭐++215. 数组中的第K个最大元素++------题解思路](#⭐215. 数组中的第K个最大元素——题解思路)
  • [3- ACM实现](#3- ACM实现)


1- 思路

快速选择

  • 第 k 大的元素的数组下标: int target = nums.length - k

1- 根据 partition 分割的区间来判断当前处理方式

  • 如果返回的 int 等于 target 说明找到了,直接返回
  • 如果返回的 int 小于 target 说明要在当前区间的右侧寻找,也就是 [pivotIndex+1,right]
  • 如果返回的 int 大于 target 说明要在当前区间的左侧寻找,也就是 [left,pivotIndex-1]

2- 实现 partition 随机选取一个 pivotIndex 分割区间

  • 2-1 随机选择一个下标
  • 2-2 交换 left 和 随机下标
  • 2-3 将随机下标的元素值设置为 pivot
  • 2-4 定义 lege 下标 使用 while(true)
    • 使得 le 指向的元素始终小于 pivot
    • 使得 ge 指向的元素始终大于 pivot

2- 实现

⭐++215. 数组中的第K个最大元素++------题解思路

java 复制代码
import java.util.Random;
class Solution {


    static Random random = new Random(System.currentTimeMillis());
    public int findKthLargest(int[] nums,int k){
        return quickSelect(nums,0,nums.length-1,nums.length-k);

    }

    public int quickSelect(int[] nums,int left,int right,int kIndex){
        if(right==left){
            return nums[left];
        }
        //
        int pivotIndex = partition(nums,left,right);

        if(pivotIndex == kIndex){
            return nums[kIndex];
        }else if( pivotIndex>kIndex){
            return quickSelect(nums,left,pivotIndex-1,kIndex);
        }else{
            return quickSelect(nums,pivotIndex+1,right,kIndex);
        }
    }

    public int partition(int[] nums,int left,int right){
        int randomIndex = left + random.nextInt(right-left+1);
        swap(nums,left,randomIndex);

        int mid = nums[left];
        int le = left+1;
        int ge = right;
        while(true){
            while(le<=ge && nums[le] < mid){
                le++;
            }
            while(le<=ge && nums[ge] > mid){
                ge--;
            }
            if(le>=ge){
                break;
            }
            swap(nums,le,ge);
            le++;
            ge--;
        }
        swap(nums,left,ge);
        return ge;
    }

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


}

3- ACM实现

java 复制代码
public class kthNums {

    static Random random = new Random(System.currentTimeMillis());
    public static int findK(int[] nums,int k){
        // 快速选择 ,传四个参数
        return quickSelect(nums,0,nums.length-1,nums.length-k);

    }

    public static int quickSelect(int[] nums,int left,int right,int kIndex){
        if(right==left){
            return nums[left];
        }
        //
        int pivotIndex = partition(nums,left,right);

        if(pivotIndex == kIndex){
            return nums[kIndex];
        }else if( pivotIndex>kIndex){
            return quickSelect(nums,left,pivotIndex-1,kIndex);
        }else{
            return quickSelect(nums,pivotIndex+1,right,kIndex);
        }
    }

    public static int partition(int[] nums,int left,int right){
        int randomIndex = left + random.nextInt(right-left+1);
        swap(nums,left,randomIndex);

        int mid = nums[left];
        int le = left+1;
        int ge = right;
        while(true){
            while(le<=ge && nums[le] < mid){
                le++;
            }
            while(le<=ge && nums[ge] > mid){
                ge--;
            }
            if(le>=ge){
                break;
            }
            swap(nums,le,ge);
            le++;
            ge--;
        }
        swap(nums,left,ge);
        return ge;
    }

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

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String input = sc.nextLine();
        String[] parts = input.split(" ");
        int[] nums = new int[parts.length];
        for(int i = 0 ; i < nums.length ; i++){
            nums[i] = Integer.parseInt(parts[i]);
        }
        System.out.println("输入K");
        int k = sc.nextInt();
        System.out.println("结果是"+findK(nums,k));

    }
}
相关推荐
知来者逆21 分钟前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊26 分钟前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北27 分钟前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈1 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing1 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD2 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘3 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV3 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8244 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify4 小时前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法