zoukankan      html  css  js  c++  java
  • 703. Kth Largest Element in a Stream

    Design a class to find the kth largest element in a stream. Note that it is the kth largest element in the sorted order, not the kth distinct element.

    Your KthLargest class will have a constructor which accepts an integer k and an integer array nums, which contains initial elements from the stream. For each call to the method KthLargest.add, return the element representing the kth largest element in the stream.

    Example:

    int k = 3;
    int[] arr = [4,5,8,2];
    KthLargest kthLargest = new KthLargest(3, arr);
    kthLargest.add(3);   // returns 4
    kthLargest.add(5);   // returns 5
    kthLargest.add(10);  // returns 5
    kthLargest.add(9);   // returns 8
    kthLargest.add(4);   // returns 8
    

    Note: 
    You may assume that nums' length ≥ k-1 and k ≥ 1.

    建立一个min heap,只包含k个最大的元素。

    constructor把nums各项放入min heap,只保留k个最大数。add函数先把当前val放入min heap,如果heap的size > k就移出堆顶最小元素。最后返回min heap的堆顶元素

    *注意记得k的处理

    时间O(NlogK),空间O(K)

    class KthLargest {
        PriorityQueue<Integer> minHeap;
        int k;
    
        public KthLargest(int k, int[] nums) {
            this.k = k;
            minHeap = new PriorityQueue<>();
            for(int n : nums) {
                minHeap.offer(n);
                if(minHeap.size() > k)
                    minHeap.poll();
            }
        }
        
        public int add(int val) {
            minHeap.offer(val);
            if(minHeap.size() > k)
                minHeap.poll();
            return minHeap.peek();
        }
    }
  • 相关阅读:
    208. Implement Trie (Prefix Tree)
    97. Interleaving String
    314. Binary Tree Vertical Order Traversal
    windows获取IP和MAC地址【Qt】
    阳历阴历转换
    getDat(char *val)获得某一天是这一年中的第几天
    int位数的获取及int类型转char *
    以二进制形式输出char *数据
    char类型变量二进制形式输出
    int类型变量以二进制形式输出
  • 原文地址:https://www.cnblogs.com/fatttcat/p/9988128.html
Copyright © 2011-2022 走看看