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

滑动窗口中位数

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

相关推荐

03-19 18:27
已编辑
门头沟学院 C++
26学院本太难了,很多公司机筛就给我刷了。机会都难拿到如果是简历存在问题也欢迎拷打————————————————————分割线——————————————————————2026.3.4更新:发完贴之后,时不时投递又收到了不少的笔试/面试邀请。主要是之前投递简历出去之后基本上都是沉默状态,年后好转了不少timeline:2026.01.21&nbsp;文远知行笔试,半年多没刷算法题&nbsp;-&gt;挂&nbsp;(后续HR说春招可以重新安排笔试)2026.2.4&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;小鹏汇天&nbsp;技术一面,第二周收到结果&nbsp;-&gt;挂2026.2.12&nbsp;&nbsp;&nbsp;大众Cariad代招&nbsp;技术二面&nbsp;-&gt;Offer2026.2.28&nbsp;&nbsp;&nbsp;多益网络技术面试,由于风评太差,一直在犹豫要不要接面试&nbsp;-&gt;推迟-----------分割线-----------2026.3&nbsp;月前的某一天,临时去电网报名了二批计算机岗位的笔试2026.3.6&nbsp;从上家公司实习离职,氛围最好的一家公司,leader&nbsp;说可以帮忙转正,但是流程太长,而且我们部门据说只有一个&nbsp;hc,更想要研究生,我很有可能是会被签外包公司在这里干活,就离职了。2026.3.9&nbsp;入职新公司,大众Cariad&nbsp;以外部公司的身份进组,项目组签了三年,后续三年应该都可以在这里呆,不知道有没有希望原地跳槽。2026.3.10&nbsp;电网考试居然说我通过资格审查了,短信约我去参加资格审查,请假一天,买了&nbsp;12&nbsp;号晚上的机票回成都2026.3.15&nbsp;参加国家电网三新计算机类的笔试2026.3.17&nbsp;电网出成绩了,感觉很低。觉得已经🈚️了2026.3.18&nbsp;收到电网面试通知,通知&nbsp;3.22-3.25&nbsp;这个时间去面试,我的岗位只招&nbsp;1&nbsp;个人。据说面试只有&nbsp;2-3&nbsp;人,不知道能不能成功
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
正在热议
更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务