Acwing154滑动窗口

题目

给定一个大小为 n≤10^6 的数组。

有一个大小为 k 的滑动窗口,它从数组的最左边移动到最右边。

你只能在窗口中看到 k 个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子:

该数组为 [1 3 -1 -3 5 3 6 7],k 为 33。

窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7

你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。

输入格式

输入包含两行。

第一行包含两个整数 n 和 k,分别代表数组长度和滑动窗口的长度。

第二行有 n 个整数,代表数组的具体数值。

同行数据之间用空格隔开。

输出格式

输出包含两个。

第一行输出,从左至右,每个位置滑动窗口中的最小值。

第二行输出,从左至右,每个位置滑动窗口中的最大值。

输入样例:

复制代码
8 3
1 3 -1 -3 5 3 6 7

输出样例:

diff 复制代码
-1 -3 -3 -3 3 3
3 3 5 5 6 7

代码与思路

java 复制代码
import java.io.*;

public class Main {
    static BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));  // 用于读取输入数据的缓冲读取器
    static BufferedWriter log = new BufferedWriter(new OutputStreamWriter(System.out));  // 用于输出结果的缓冲写入器

    public static void main(String[] args) throws IOException {
        // 对输入数据进行初始化
        String[] s1 = reader.readLine().split(" ");
        int n = Integer.parseInt(s1[0]);  // 数组长度
        int k = Integer.parseInt(s1[1]);  // 滑动窗口大小
        String[] s = reader.readLine().split(" ");
        int[] a = new int[n];  // 输入数组
        for (int i = 0; i < n; i++) {
            a[i] = Integer.parseInt(s[i]);  // 初始化输入数组
        }
        int[] ans = new int[n];  // 存储滑动窗口中的最小(或最大)值的下标
        int tt = -1, hh = 0;  // 初始化队列的头尾指针

        // 查找滑动窗口中的最小值
        for (int i = 0; i < n; i++) {
            if (hh <= tt && i - k + 1 > ans[hh]) {
                hh++;
            }
            while (hh <= tt && a[i] <= a[ans[tt]]) {
                tt--;
            }
            ans[++tt] = i;
            if (i + 1 >= k) {
                log.write(a[ans[hh]] + " ");  // 输出滑动窗口中的最小值
            }
        }

        log.write("\n");  // 换行

        tt = -1;  // 重新初始化队列的尾指针
        hh = 0;  // 重新初始化队列的头指针
        // 查找滑动窗口中的最大值
        for (int i = 0; i < n; i++) {
            if (hh <= tt && i - k + 1 > ans[hh]) {
                hh++;
            }
            while (hh <= tt && a[i] >= a[ans[tt]]) {
                tt--;
            }
            ans[++tt] = i;
            if (i + 1 >= k) {
                log.write(a[ans[hh]] + " ");  // 输出滑动窗口中的最大值
            }
        }

        // 关闭输入输出流
        log.flush();
        reader.close();
        log.close();
    }
}
相关推荐
im_AMBER4 分钟前
Web 开发 30
前端·笔记·后端·学习·web
码事漫谈14 分钟前
LLVM IR深度技术解析:架构、优化与应用
后端
码事漫谈17 分钟前
C++ 中的类型转换:深入理解 static_cast 与 C风格转换的本质区别
后端
小蒜学长37 分钟前
springboot餐厅信息管理系统设计(代码+数据库+LW)
java·数据库·spring boot·后端
Chh432241 小时前
React 新版
后端
Miracle6581 小时前
【征文计划】Rokid CXR-M SDK全解析:从设备连接到语音交互的AR协同开发指南
后端
合作小小程序员小小店2 小时前
web开发,学院培养计划系统,基于Python,FlaskWeb,Mysql数据库
后端·python·mysql·django·web app
笃行3502 小时前
基于Rokid CXR-S SDK的智能AR翻译助手技术拆解与实现指南
后端
文心快码BaiduComate2 小时前
代码·创想·未来——百度文心快码创意探索Meetup来啦
前端·后端·程序员
渣哥2 小时前
面试官最爱刁难:Spring 框架里到底用了多少经典设计模式?
javascript·后端·面试