ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

leetcode (堆->simple)703,1046,前k大/小数

2021-02-02 16:32:58  阅读:174  来源: 互联网

标签:1046 k2 simple k1 int add heap poll leetcode


703

第K大

 

 

class KthLargest {

    private PriorityQueue<Integer> heap ;
    
    private int k;

    public KthLargest(int k, int[] nums) {
heap = new PriorityQueue<>(k,(k1,k2)->k1.compareTo(k2));
        this.k = k;
        for (int i = 0; i < nums.length; i++) {
            if (i < k){
                heap.add(nums[i]);
            }else {
                if (nums[i] >= heap.peek()){
                    heap.poll();
                    heap.add(nums[i]);
                }
            }
        }
    }
    
    public int add(int val) {
        if (heap.size() < k){
            heap.add(val);
        } else if (val >= heap.peek()){
            heap.poll();
            heap.add(val);
        }
        return heap.peek();
    }
}

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */

 

1046

 

 

也是用堆的特性来做

        PriorityQueue<Integer> heap = new PriorityQueue<>((k1,k2)->k2.compareTo(k1));
        for (int i = 0; i < stones.length; i++) {
            heap.offer(stones[i]);
        }

        while (heap.size() > 1){
            Integer x1 = heap.poll();
            Integer x2 = heap.poll();
            if (x1 != x2){
                heap.offer(Math.abs(x1-x2));
            }
        }
        if (heap.isEmpty()){
            return 0;
        }

        return heap.poll();

 

 

前大/小K 个数

    public static int[] getLeastNumbers(int[] arr,int k) {
        if (k ==0 || arr.length == 0){
            return new int[0];
        }
        PriorityQueue<Integer> heap = new PriorityQueue<>((k1,k2)->k2.compareTo(k1));
        for (int i = 0; i < arr.length; i++) {
            if (i < k){
                heap.offer(arr[i]);
            }else {
                if (arr[i] < heap.peek()){
                    heap.poll();
                    heap.offer(arr[i]);
                }
            }
        }
        int size = heap.size(),ks = 0;
        int[] result = new int[size];
        while (ks<size){
            result[ks++] = heap.poll();
        }

        return result;
    }

 

标签:1046,k2,simple,k1,int,add,heap,poll,leetcode
来源: https://www.cnblogs.com/hetutu-5238/p/14345189.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有