题解 | #查找第K小数#

查找第K小数

https://www.nowcoder.com/practice/204dfa6fcbc8478f993d23f693189ffd

采用优先队列,元素乘-1再入队,取巧地在优先队列默认大根堆的情况下实现了小根堆,最后别忘乘回来-1即可。

第k小,但相同的元素算同1个,也就是只有所有相同元素都从优先队列pop出去后,才i++。

#include <queue>
#include <cstdio>
#include <iostream>

using namespace std;

int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        priority_queue<int> myPriorityQueue;
        for(int i = 0;i<n;i++){
            int num;
            cin>>num;
            myPriorityQueue.push(num*(-1));
        }

        int k;
        cin>>k;
        for(int i = 0;i<k-1;){//把第k小之前的都出队
            //但是要实现相同大小算一样大,所以要前后比较,分情况讨论
            int curnum1 = myPriorityQueue.top();
            myPriorityQueue.pop();
            int curnum2 = myPriorityQueue.top();
            if(curnum1 == curnum2){
                //如果当前top出的元素与下一个队首元素=,则i不增加
                continue;
            }
            else{
                //只有当当前top出的元素已经是与它相同的最后一个,再i++
                i++;
            }
        }
        int answer =  myPriorityQueue.top()*(-1);
        printf("%d\n",answer);
    }
}
全部评论

相关推荐

小浪_Coding:1. 个人技能排版太乱, 写的技术栈太浅了, 跟测试,自动化相关的太少; 2. 项目开发类的太简单没有亮点, 算法类的项目建议只放一个,最好有自动化,CI/CD, pipline的项目, 需要更换; 3.整体排版需要优化, SOOB打招呼都需要注意等.
我的简历长这样
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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