程序员代码面试指南第二版 7.生成窗口最大值数组

welcome to my blog

程序员代码面试指南第二版 7.生成窗口最大值数组

题目描述

题目描述
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边，窗口每次向右边滑一个位置，
求每一种窗口状态下的最大值。（如果数组长度为n，窗口大小为w，则一共产生n-w+1个窗口的最大值）

输入描述:
第一行输入n和w，分别代表数组长度和窗口大小
第二行输入n个整数X_i，表示数组中的各个元素

输出描述:
输出一个长度为n-w+1的数组res，res[i]表示每一种窗口状态下的最大值

示例1
输入
8 3
4 3 5 4 3 3 6 7

输出
5 5 5 4 6 7

第一次做, 用队列保存当前滑窗内最大值的索引; 注意如何判断队列首元素是否在滑窗内

import java.util.Scanner;
import java.util.LinkedList;

public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] s = sc.nextLine().split(" ");
        int size = Integer.valueOf(s[1]);
        String[] array = sc.nextLine().split(" ");
        int[] arr = new int[array.length];
        for(int i=0; i<arr.length; i++)
            arr[i] = Integer.valueOf(array[i]);
        int[] res = Core(arr, size);
        for(int i=0; i<res.length; i++)
            System.out.print(res[i]+" ");
    }
    public static int[] Core(int[] arr, int size){
        //队列头元素是当前滑窗最大值对应的索引
        LinkedList<Integer> ll = new LinkedList<>();
        int[] res = new int[arr.length - size + 1];
        int index=0;
        for(int i=0; i<arr.length; i++){
            if(ll.isEmpty())
                ll.add(i);
            else{
                while(!ll.isEmpty() && arr[i] > arr[ll.getLast()]){
                    ll.removeLast();
                }
                ll.add(i);
                
                if(ll.getLast() - ll.getFirst() + 1> size)
                    ll.removeFirst();
            }
            if(i+1 >= size){
                //保存结果时要保存索引对应的值
                res[index++] = arr[ll.getFirst()];
            }
        }
        return res;
    }
}