LeetCode215 数组中的第K个最大元素

本文介绍如何使用Python实现LeetCode上的Kth Largest Element in an Array问题,通过选择排序算法解决,分析了时间复杂度为O(n^2)。重点在于代码实现和排序原理的理解。
摘要由CSDN通过智能技术生成

链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

题目分析

这道题,实际上就是考排序算法。对数组从大到小排序完,返回 nums[k-1] 。
排序方法有很多,就像那十大排序法。

官方题解提供了快速排序和堆排序两种方案,时间和空间复杂度各有优势:

排序算法时间复杂度空间复杂度
快速排序O(n)O(logn)
堆排序O(nlogn)O(logn)

我这里使用了选择排序。它有个特点是,
经过第1次遍历,第1大的元素就在 nums[0];
经过第2次遍历,第2大的元素就在 nums[1];

经过第k次遍历,第k大的元素在 nums[k-1]

代码实现

class Solution {
    public int findKthLargest(int[] nums, int k) {
        for (int i = 0; i < nums.length; i++) {
            int maxIndex = i;
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[j] > nums[maxIndex]) {
                    maxIndex = j; // 找到最大元素的下标
                }
            }
            // 交换元素
            int tmp = nums[i];
            nums[i] = nums[maxIndex];
            nums[maxIndex] = tmp;

            // 第i次结束,实际上第 i+1 大的元素就出来了,后面的就不排了
            if (i + 1 == k) {
                break;
            }
        }
        return nums[k-1];
    }
}

时间复杂度:最坏情况当 k 等于 n 时,为 O(n^2)。k = 1时,O(n)
空间复杂度:因为没申请额外数组,O(1)
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值