题解 | #滑动窗口中位数#

滑动窗口中位数

https://www.nowcoder.com/practice/820366502ed6451eb9eebb5b129ed9fa?tpId=196&tqId=40526&rp=1&ru=/exam/oj&qru=/exam/oj&sourceUrl=%2Fexam%2Foj%3Fpage%3D1%26pageSize%3D50%26search%3D%25E6%25BB%2591%25E5%258A%25A8%25E7%25AA%2597%25E5%258F%25A3%26tab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D196&difficulty=undefined&judgeStatus=undefined&tags=&title=%E6%BB%91%E5%8A%A8%E7%AA%97%E5%8F%A3

思路: 自己维护一个窗口,使窗口里面的数字是有序的。如果能够做到这一点,那么我们每次取中位数就相当方便了。大致思路有了之后,我们应该解决下面问题

  1. 窗口应该使用什么数据结构? 在这里,我们使用 ArrayList,选取的理由主要是查找速度相对于 LinkedList 更快。我们需要频繁地插入数据,每次数据插入的位置,我们都要进行 检索,使其插入合适的位置,才能保证窗口内的数据一直保持有序。
  2. 每获取到一个数据后,如何在窗口内找到插入的位置? 这里我们使用 二分查找(O(logN))。在查找过程中,如果某个位置上的值 大于等于 当前值,直接在该位置上插入;如果小于,往后一位再插入。
  3. 由于我们使用 ArrayList,因此,每一次插入的时间复杂度,都是 O(n)。 我们每一次插入,当前位置往后的数据都要进行移动,所以时间复杂度为 O(n)。

这里有几点需要主要:

  1. 防止数据溢出。 (虽然在当前题目中,原始数组中的每个值都不会取到最大。) 因此在求,中位数为两个数的平均值时,要特别注意。
  2. 精度的把握。 题目要求的返回值是 double 类型,需要注意。

最后,当前代码的用力通过率是 80%,有两个没有通过,超时了! 代码应该还能优化。这个代码我放在 Leetcode 上是能通过的,而且时间战胜了 30%多 的代码,在 牛客 上不知道为什么超时了,哎。

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型ArrayList 
     * @param k int整型 
     * @return double浮点型ArrayList
     */
    public ArrayList<Double> slidewindow (ArrayList<Integer> nums, int k) {
        // write code here
        if (null == nums || nums.size() < 1 || nums.size() < k || k == 0) {
            return new ArrayList<Double>();
        }
        ArrayList<Double> ans = new ArrayList<>();
        if (k == 1) {
            for (int num : nums) {
                ans.add((double) num);
            }
            return ans;
        }
        ArrayList<Integer> sw = new ArrayList<>();
        for (int i = 0; i < nums.size(); i++) {
            if (sw.size() == 0) {
                sw.add(nums.get(i));
                continue;
            }
            int insertIndex = binarySearch(sw, nums.get(i));
            sw.add(insertIndex, nums.get(i));
            if (sw.size() > k) {
                sw.remove(Integer.valueOf(nums.get(i - k)));
            }
            if (sw.size() == k) {
                if (k % 2 == 0) {
                    int p1 = sw.size() / 2 - 1;
                    int p2 = sw.size() / 2;
                    int val1 = sw.get(p1);
                    int val2 = sw.get(p2);
                    ans.add(val1 + ((double) val2 - val1) / 2);
                } else {
                    int p = sw.size() / 2;
                    ans.add((double) sw.get(p));
                }
            }
        }
        return ans;
    }
    public static int binarySearch(ArrayList<Integer> sw, int target) {
        int l = 0;
        int r = sw.size() - 1;
        while (l < r) {
            int mid = l + ((r - l) >> 1);
            if (sw.get(mid) < target) {
                l = mid + 1;
            } else if (sw.get(mid) > target) {
                r = mid - 1;
            } else {
                return mid;
            }
        }
        return sw.get(l) > target ? l : l + 1;
    }
}
全部评论
该牛油正在参与牛客写题解薅羊毛的活动,牛币,周边,京东卡超多奖品放送,活动进入倒计时!快来捡漏啦https://www.nowcoder.com/discuss/888949?source_id=profile_create_nctrack&channel=-1
点赞 回复 分享
发布于 2022-04-20 17:12

相关推荐

伟大的烤冷面被普调:暨大✌🏻就是强
点赞 评论 收藏
分享
评论
点赞
收藏
分享
正在热议
# 25届秋招总结 #
440109次浏览 4488人参与
# 春招别灰心,我们一人来一句鼓励 #
41383次浏览 524人参与
# 阿里云管培生offer #
119676次浏览 2219人参与
# 地方国企笔面经互助 #
7914次浏览 18人参与
# 虾皮求职进展汇总 #
113709次浏览 881人参与
# 实习,投递多份简历没人回复怎么办 #
2453743次浏览 34846人参与
# 北方华创开奖 #
107254次浏览 599人参与
# 实习必须要去大厂吗? #
55563次浏览 960人参与
# 同bg的你秋招战况如何? #
75265次浏览 549人参与
# 提前批简历挂麻了怎么办 #
149784次浏览 1977人参与
# 投递实习岗位前的准备 #
1195605次浏览 18546人参与
# 你投递的公司有几家约面了? #
33166次浏览 188人参与
# 双非本科求职如何逆袭 #
661802次浏览 7394人参与
# 机械人春招想让哪家公司来捞你? #
157587次浏览 2267人参与
# 如果公司给你放一天假,你会怎么度过? #
4717次浏览 54人参与
# 如果你有一天可以担任公司的CEO,你会做哪三件事? #
11266次浏览 263人参与
# 发工资后,你做的第一件事是什么 #
12368次浏览 61人参与
# 工作中,努力重要还是选择重要? #
35546次浏览 384人参与
# 参加完秋招的机械人,还参加春招吗? #
20072次浏览 240人参与
# 实习想申请秋招offer,能不能argue薪资 #
39211次浏览 314人参与
# 我的上岸简历长这样 #
451881次浏览 8088人参与
# 非技术岗是怎么找实习的 #
155832次浏览 2120人参与
牛客网
牛客企业服务