滑动窗口

来源:互联网 发布:淘宝双11销量排名 编辑:程序博客网 时间:2024/05/17 20:08

有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7],w=3为例。因为第一个窗口[4,3,5]的最大值为5,第二个窗口[3,5,4]的最大值为5,第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。

给定整形数组arr及它的大小n,同时给定w,请返回res数组。保证w小于等于n,同时保证数组大小小于等于500。

测试样例:

[4,3,5,4,3,3,6,7],8,3

返回:[5,5,5,4,6,7]

思路:每一次循环遍历W个数字用i到i+w即可

代码如下:

import java.util.*;public class SlideWindow {  public static int[] slide(int[] arr, int n, int w) {int[] res = new int[n - w + 1];for (int i = 0; i <= n - w; i++) {res[i] = getMax(arr, i, i + w);}return res;}public static int getMax(int[] arr, int start, int end) {int max = 0;for (int i = start; i < end; i++) {if (max < arr[i])max = arr[i];}return max;}}




0 0
原创粉丝点击