480. 滑动窗口中位数

最新推荐文章于 2024-05-23 22:02:44 发布

LIZHUOLONG1

最新推荐文章于 2024-05-23 22:02:44 发布

阅读量45

点赞数

分类专栏： LeetCode刷题文章标签：算法

本文链接：https://blog.csdn.net/LIZHUOLONG1/article/details/134511168

版权

LeetCode刷题专栏收录该内容

304 篇文章 4 订阅

订阅专栏

480. 滑动窗口中位数

在这里插入图片描述

Java：超出时间限制

class Solution {
    double[] ans;
    int j;

    private void sortAndSetAns(int k, Map<Integer, Integer> map) {
        List<Map.Entry<Integer,Integer>> list = new ArrayList<>(map.entrySet());
        Collections.sort(list, (o1, o2) -> o1.getValue().compareTo(o2.getValue()));
        if (k % 2 == 1) {
            ans[j++] = list.get(k/2).getValue();
        } else {
            ans[j++] = list.get(k/2).getValue() / 2.0 + list.get(k/2 - 1).getValue() / 2.0;
        }
        list.clear();
    }

    public double[] medianSlidingWindow(int[] nums, int k) {
        ans = new double[nums.length - k + 1];
        j = 0;

        Map<Integer, Integer> map = new HashMap<>();
        for(int i = 0; i < k; ++i) {
            map.put(i, nums[i]);
        }
        sortAndSetAns(k, map);

        for (int i = k; i < nums.length; ++i) {
            map.remove(i-k, map.get(i-k));
            map.put(i, nums[i]);
            sortAndSetAns(k, map);
        }

        return ans;
    }
}