题解 | 【模板】滑动窗口

【模板】滑动窗口

https://www.nowcoder.com/practice/be419f584a3f4c5b818833f1ce856626

import java.util.*;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int k = in.nextInt();
        int[] nums = new int[n];
        int index = 0;
        // 注意 hasNext 和 hasNextLine 的区别
        while (in.hasNextInt()) { // 注意 while 处理多个 case
            nums[index] = in.nextInt();
            index++;
        }
        // 单调队列在处理滑动窗口时有高效率
        // 双端队列:适合处理滑动状况这种需要移除队尾元素的情况
        Deque<Integer> dq = new LinkedList<>(); // 存储下标
        ArrayList<Integer> result = new ArrayList<>();
        // 维护一个单调递减序列
        for(int i = 0; i < n; i++){
            // 移除所有小于当前元素的队尾下标,维护最大值是队头元素
            while(!dq.isEmpty() && nums[dq.peekLast()] <= nums[i]){
                dq.pollLast();
            }
            // 将当前元素下标入队
            dq.offerLast(i);
            if(dq.peekFirst() <= i - k){
                dq.pollFirst();
            }
            // 当窗口形成后,记录最大值
            if(i >= k - 1){
                result.add(nums[dq.peekFirst()]);
            }
        }
        for(int x : result){
            System.out.print(x + " ");
        }

    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
10-09 17:17
已编辑
门头沟学院 Java
活泼的代码渣渣在泡池...:同学你好,我也是学院本,后天要面这个亚信科技,是实习,请问问题都啥样呀,我项目就做了网上的,这是第一次面试
投递多益网络等公司10个岗位
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

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