标签:le 15 iterator res 元素 queue NC119 2022 input
描述
给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。 数据范围:0\le k,n \le 100000≤k,n≤10000,数组中每个数的大小0 \le val \le 10000≤val≤1000 要求:空间复杂度 O(n)O(n) ,时间复杂度 O(nlogn)O(nlogn)方法:优先队列、迭代器、lambda表达式
import java.util.*; public class Solution { //堆的使用、lambda表达式的使用、迭代器的使用 public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) { ArrayList<Integer> res = new ArrayList<>(); if(input.length == 0 || k == 0) { return res; } //怎么判断是大根堆还是小根堆,目的是大根堆 PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b - a); for(int i = 0; i < input.length; i++) { if(queue.size() < k) { queue.add(input[i]); } else if(queue.peek() > input[i]) { queue.poll(); queue.add(input[i]); } } Iterator<Integer> iterator = queue.iterator(); while(iterator.hasNext()) { res.add(iterator.next()); } return res; } }
要保证堆的大小不能超过K,然后遍历数组,因为是最大堆,也就是堆顶元素是堆中最大的,如果遍历的元素小于堆顶元素,就把堆顶元素给移除,然后再把当前遍历的元素加入到堆中,最后在把堆中元素转化为数组即可。
怎么判断是大根堆:(a,b) -> b - a,小-大
标签:le,15,iterator,res,元素,queue,NC119,2022,input 来源: https://www.cnblogs.com/liujinhui/p/15804923.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。