题解 | #第k小#

第k小

https://ac.nowcoder.com/acm/problem/214362

类似于求中位数,求中位数是将前一半小的数放入大顶堆,而本题可以前k小的数放入大顶堆中,因此维护大小始终为k的大顶堆即可。

对于准备插入的数

  1. 如果size小于k就直接插入

  2. 大于等于堆顶就跳过

  3. 小于堆顶就踢出当前堆顶,插入

import java.io.*;
import java.util.*;

public class Main {

	public static void main(String[] args) {
		InputReader in = new InputReader();
		PrintWriter out = new PrintWriter(System.out);
        PriorityQueue<Integer> p = new PriorityQueue<>();
        int n = in.nextInt();
        int m = in.nextInt();
        int k = in.nextInt();
        for(int i = 0; i < n; ++i) {
            int num = in.nextInt();
            if(p.size() < k)
                p.add(-num);
            else if(num < -p.peek()) {
                p.poll();
                p.add(-num);
            }
        }
        for(int i = 0; i < m; ++i) {
            int opr = in.nextInt();
                if(opr == 1) {
                int num = in.nextInt();
                if(p.size() < k)
                    p.add(-num);
                else if(num < -p.peek()) {
                    p.poll();
                    p.add(-num);
                }
            } else out.println(p.size() < k ? -1 : -p.peek());
        }
        
		out.close();
	}
}

全部评论

相关推荐

勤奋努力的椰子这就开摆:美团骑手在美团工作没毛病
投递美团等公司10个岗位
点赞 评论 收藏
分享
尊尼获获:闺蜜在哪?
点赞 评论 收藏
分享
评论
5
收藏
分享
正在热议
# 25届秋招总结 #
442065次浏览 4508人参与
# 春招别灰心,我们一人来一句鼓励 #
41866次浏览 531人参与
# 阿里云管培生offer #
120196次浏览 2219人参与
# 地方国企笔面经互助 #
7957次浏览 18人参与
# 同bg的你秋招战况如何? #
76477次浏览 561人参与
# 虾皮求职进展汇总 #
115376次浏览 886人参与
# 北方华创开奖 #
107421次浏览 599人参与
# 实习,投递多份简历没人回复怎么办 #
2454553次浏览 34856人参与
# 实习必须要去大厂吗? #
55760次浏览 961人参与
# 提前批简历挂麻了怎么办 #
149886次浏览 1977人参与
# 投递实习岗位前的准备 #
1195903次浏览 18548人参与
# 你投递的公司有几家约面了? #
33203次浏览 188人参与
# 双非本科求职如何逆袭 #
662154次浏览 7394人参与
# 如果公司给你放一天假,你会怎么度过? #
4750次浏览 55人参与
# 机械人春招想让哪家公司来捞你? #
157622次浏览 2267人参与
# 如果你有一天可以担任公司的CEO,你会做哪三件事? #
11525次浏览 284人参与
# 发工资后,你做的第一件事是什么 #
12659次浏览 62人参与
# 工作中,努力重要还是选择重要? #
35779次浏览 384人参与
# 参加完秋招的机械人,还参加春招吗? #
20120次浏览 240人参与
# 我的上岸简历长这样 #
451995次浏览 8088人参与
# 实习想申请秋招offer,能不能argue薪资 #
39286次浏览 314人参与
# 非技术岗是怎么找实习的 #
155864次浏览 2120人参与
牛客网
牛客企业服务