zoukankan      html  css  js  c++  java
  • Sliding Window Median

    Median is the middle value in an ordered integer list. If the size of the list is even, there is no middle value. 
    So the median is the mean of the two middle value. Examples: [2,3,4] , the median is 3 [2,3], the median is (2 + 3) / 2 = 2.5 Given an array nums, there is a sliding window of size k which is moving from the very left of the array to the very right.
    You can only see the k numbers in the window. Each time the sliding window moves right by one position.
    Your job is to output the median array for each window in the original array. For example, Given nums = [1,3,-1,-3,5,3,6,7], and k = 3. Window position Median --------------- ----- [1 3 -1] -3 5 3 6 7 1 1 [3 -1 -3] 5 3 6 7 -1 1 3 [-1 -3 5] 3 6 7 -1 1 3 -1 [-3 5 3] 6 7 3 1 3 -1 -3 [5 3 6] 7 5 1 3 -1 -3 5 [3 6 7] 6 Therefore, return the median sliding window as [1,-1,-1,3,5,6].

    方法1:Time Complexity O(NK)

    暂时只有两个Heap的做法,缺点:In this problem, it is necessary to be able remove elements that are not necessarily at the top of the heap. PriorityQueue has logarithmic time remove top, but a linear time remove arbitrary element.

    For a Heap:

    remove():  Time Complexity is O(logN)

    remove(Object): Time Complexity is O(N)

    更好的有multiset的方法,但是还没有看到好的java version的

    最大堆的简单定义方法:Collections.reverseOrder(), Returns a comparator that imposes the reverse of the natural ordering on a collection of objects

    public class Solution {
        PriorityQueue<Double> high = new PriorityQueue();
        PriorityQueue<Double> low = new PriorityQueue(Collections.reverseOrder());
        
        
        public double[] medianSlidingWindow(int[] nums, int k) {
            double[] res = new double[nums.length-k+1];
            int index = 0;
    
            for (int i=0; i<nums.length; i++) {
                if (i >= k) remove(nums[i-k]);
                add((double)nums[i]);
                if (i >= k-1) {
                    res[index++] = findMedian();
                }
            }
            return res;
        }
        
        public void add(double num) {
            low.offer(num);
            high.offer(low.poll());
            if (low.size() < high.size()) {
                low.offer(high.poll());
            }
        }
        
        public double findMedian() {
            if (low.size() == high.size()) {
                return (low.peek() + high.peek()) / 2.0;
            }
            else return low.peek();
        }
        
        public void remove(double num) {
            if (num <= findMedian()) {
                low.remove(num);
            }
            else {
                high.remove(num);
            }
            if (low.size() < high.size()) {
                low.offer(high.poll());
            }
            else if (low.size() > high.size()+1) {
                high.offer(low.poll());
            }
        }
    }
    

      

  • 相关阅读:
    Windows10远程桌面连接提示:出现身份验证错误,要求的函数不受支持
    mybatis 中 if-test 判断大坑
    hutool的DateUtil工具类
    SpringBoot启动过程
    数据库事务的隔离级别
    EasyUI管理后台模板(附源码)
    springmvc中自定义拦截器以及拦截器的执行过程
    文件上传(MultipartFile)
    文件下载(使用springmvc框架中ResponseEntity对象)
    json格式实现数据传输
  • 原文地址:https://www.cnblogs.com/apanda009/p/7798455.html
Copyright © 2011-2022 走看看