题解 | #第k轻的牛牛 ,对比三种算法!#

第k轻的牛牛

https://www.nowcoder.com/practice/7676478b46794456b145e8e48b0e2763

题意:无序数组中找第k小元素,优化时间复杂度

思路:我大概想到了三种算法,各有好坏,请君评断

1.简化版桶排序,

将数组中的元素放到作为桶数组的索引,当我们完全放完的时候,遍历桶数组,当桶数组=1(标记),

拿到所有=1的数组,即一个完成的顺序。

时间复杂度 o(n)

2.最大堆+k大小维护

将数组中的元素全部放到一个k大小的堆中,进行维护,最后取堆中的最后一个元素即可

时间复杂度 o(n*logk)

3.快速选择算法

在快速排序中,算法会递归处理划分两边,但是选择排序我们只处理一边

平均时间复杂 o(n)

对比:第一种算法空间有限,只适用于数组下标遍布均匀的情况下。

第二种,复杂度略高,适用于处理动态场景,比如不仅仅是取k小,而是找一堆,

第三种,适用于找数组中k小的最佳算法,最坏情况下复杂度o2,平均on

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param weights int整型一维数组
     * @param k int整型
     * @return int整型
     */
    public int findKthSmallest (int[] weights, int k) {
        // write code here
        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b - a); //最大堆
        for (int i = 0; i < weights.length; i++) {
            if (queue.size() < k)
                queue.offer(weights[i]);
            else {
                if (weights[i] < queue.peek()) { //当元素小的才会推
                    queue.poll();
                    queue.offer(weights[i]);
                }
            }
        }
        return queue.peek();
    }
}

全部评论

相关推荐

来,说点可能被同行“骂”的大实话。🙊当初接数字马力Offer时,朋友都说:“蚂蚁的“内包”公司?你想清楚啊!”但入职快一年后的今天,我反而对他有了不一样的看法!🔹&nbsp;是偏见?还是信息差!之前没入职之前外面都在说什么岗位低人一等这类。实际上:这种情况不可至否,不能保证每个团队都是其乐融融。但我在的部门以及我了解的周边同事都还是十分好相处的~和蚂蚁师兄师姐之间也经常开一些小玩笑。总之:身份是蚂蚁公司给的,地位是自己挣的(一个傲娇女孩的自述)。🔹&nbsp;待遇?玩的就是真实!试用期工资全额发!六点下班跑得快(早9晚6或者早10晚7,动态打卡),公积金顶格交。别听那些画饼的,到手的钱和下班的时间才是真的(都是牛马何必难为牛马)。🔹&nbsp;能不能学到技术?来了就“后悔”!我们拥有权限直通蚂蚁知识库,技术栈多到学不完。说“学不到东西”的人,来了可能后悔——后悔来晚了(哈哈哈哈,可以不学但是不能没有)!💥&nbsp;内推地址:https://app.mokahr.com/su/ueoyhg❗我的内推码:NTA6Nvs走我的内推,可以直达业务部门,面试流程更快速,进度可查!今天新放HC,之前挂过也能再战!秋招已经正式开始啦~机会就摆在这,敢不敢来试一试呢?(和我一样,做个勇敢的女孩)
下午吃泡馍:数字马力的薪资一般哇,5年经验的java/测试就给人一万出头,而且刚入职第三天就让人出差,而且是出半年
帮你内推|数字马力 校招
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务