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();
    }
}
相关推荐
yu41062131 分钟前
Rust 语言使用场景分析
开发语言·后端·rust
细心的莽夫1 小时前
SpringCloud 微服务复习笔记
java·spring boot·笔记·后端·spring·spring cloud·微服务
jack_xu2 小时前
高频面试题:如何保证数据库和es数据一致性
后端·mysql·elasticsearch
pwzs3 小时前
Java 中 String 转 Integer 的方法与底层原理详解
java·后端·基础
Asthenia04123 小时前
InnoDB文件存储结构与Socket技术(从Linux的FD到Java的API)
后端
Asthenia04123 小时前
RocketMQ 消息不丢失与持久化机制详解-生产者与Broker之间的详解
后端
〆、风神4 小时前
Spring Boot 整合 Lock4j + Redisson 实现分布式锁实战
spring boot·分布式·后端
Asthenia04124 小时前
Select、Poll、Epoll 详细分析与面试深度剖析/C代码详解
后端
烛阴4 小时前
Node.js中必备的中间件大全:提升性能、安全与开发效率的秘密武器
javascript·后端·express