滑动窗口求最大值数组

【题目】给出一个整形数组,例如arr = {5,4,3,5,6,7,6},窗口大小为w=3,窗口每次向右移动一位,输出每个窗口中最大值组成的数组。

package string;

import com.google.common.collect.Lists;

import java.util.Deque;
import java.util.LinkedList;
import java.util.Random;

/**
 * Created by cqs on 2018/3/17.
 */
public class MaxArray {

    private static Random random = new Random();


    public static void main(String[] args) {
//        int[] a = {4, 3, 5, 4, 3, 3, 6, 7};
        Integer[] a = {5, 4, 3, 5, 6, 7, 6};
        a = randArr(10);
        int windows = 3;
        Integer[] result = getMaxArray(a, windows);
        System.out.println(Lists.newArrayList(result));

    }

    /**
     * 使用双端队列存储数组索引
     * 队首是当前最大的值
     * <p>
     * <p>
     * 遍历数组的时候 当下一个数的值小于队尾时候添加到队列尾部
     * 当下一个数X大于等于队尾时 删除队尾元素  并继续和新的队尾比较 直到不大于队尾元素或者队列为空
     * 然后添加X的索引到队列尾部
     * <p>
     * 将队列首部元素(索引)对应数组的值存储到结果数组中
     * <p>
     * 队列最多windows个元素 当队列超过windows个元素时 队首元素过期了 (窗口已经划过了) 需要删除队首元素
     */
    private static Integer[] getMaxArray(Integer[] a, int windows) {
        if (a == null || a.length < windows) return null;
        Deque<Integer> queue = new LinkedList<>();
        //处理第1个元素 至第windows -1个元素
        queue.add(0);
        for (int i = 1; i < windows - 1; i++) {
            int fIndex = queue.getFirst();
            if (a[i] > a[fIndex]) {
                queue.removeFirst();
                queue.addFirst(i);
            }
        }
        //从第windows个元素开始 每次求出一个窗口最大值
        Integer[] result = new Integer[a.length - windows + 1];
        for (int i = windows - 1; i < a.length; i++) {
            //a[i]大于等于队尾的话 删除队尾元素 直到队尾元素大于a[i] 或者队列为空
            while (queue.size() > 0 && a[queue.getFirst()] <= a[i]) {
                queue.removeLast();
            }
            //队尾添加
            queue.addLast(i);
            int fIndex = queue.getFirst();
            result[i - windows + 1] = a[fIndex];
            if (i - windows + 1 >= fIndex) { //本轮中窗口划过队首元素
                queue.removeFirst();
            }
        }
        return result;
    }

    private static Integer[] randArr(int size) {
        Integer[] as = new Integer[size];
        for (int i = 0; i < size; i++) {
            as[i] = random.nextInt(100 * size);
        }
        System.out.println(Lists.newArrayList(as));
        return as;
    }
}

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 题目 有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一...
    IT_Matters阅读 7,108评论 0 3
  • 题目 有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 例如,数组为[...
    50fc16abfd49阅读 5,023评论 0 1
  • 第2章 基本语法 2.1 概述 基本句法和变量 语句 JavaScript程序的执行单位为行(line),也就是一...
    悟名先生阅读 9,781评论 0 13
  • 不知是谁给你起的芳名 若不亲睹您国色天香的真容 会误认为您如墨染、黑炭一般 你只是深深的红、深深的紫……着装深暗 ...
    蕙兰漱雪阅读 3,153评论 8 5
  • 你看过这本书没?《做,就对了》。不像很多鸡血励志书籍那样,这本书简单直接的告诉你,有想法就付诸行动,别想太多...
    小昭妮儿阅读 1,477评论 2 2

友情链接更多精彩内容