leetcode480. 滑动窗口中位数

96 篇文章 1 订阅
81 篇文章 1 订阅

题目:
https://leetcode-cn.com/problems/sliding-window-median/comments/
就是建个数组,每次移动就是移除一个(移除前用二分查找找到具体点),加上一个。其他没难度
代码:

public double[] medianSlidingWindow(int[] nums, int k) {
    int len = nums.length;
    double[] ans = new double[len - k + 1];
    int[] sz = Arrays.copyOfRange(nums, 0, k);
    Arrays.sort(sz);
    if (k % 2 == 0) {
        for (int i = 0; i < ans.length - 1; i++) {
            ans[i] = (0.0 + sz[k / 2] + sz[k / 2 - 1]) / 2;
            if (nums[i] == nums[i + k]) {
                continue;
            }
            change(sz, nums[i], nums[i + k], k);
        }
        ans[ans.length - 1] = (0.0 + sz[k / 2] + sz[k / 2 - 1]) / 2;
    } else {
        for (int i = 0; i < ans.length - 1; i++) {
            ans[i] = sz[k / 2];
            if (nums[i] == nums[i + k]) {
                continue;
            }
            change(sz, nums[i], nums[i + k], k);
        }
        ans[ans.length - 1] = sz[k / 2];
    }
    return ans;
}

private void change(int[] sz, int remove, int add, int len) {
    int key = getKey(sz, remove, 0, len - 1);
    if (add > remove) {
        for (int i = key + 1; i < len; i++) {
            if (sz[i] < add) {
                sz[i - 1] = sz[i];
            } else {
                sz[i - 1] = add;
                return;
            }
        }
        sz[len - 1] = add;
    } else {
        for (int i = key - 1; i >= 0; i--) {
            if (sz[i] > add) {
                sz[i + 1] = sz[i];
            } else {
                sz[i + 1] = add;
                return;
            }
        }
        sz[0] = add;
    }
}

private int getKey(int[] sz, int remove, int sta, int end) {
    int middle = (sta + end) / 2;
    if (sz[middle] == remove) {
        return middle;
    }
    if (sz[middle] > remove) {
        return getKey(sz, remove, sta, middle - 1);
    } else {
        return getKey(sz, remove, middle + 1, end);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值