题目
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边划一个位置。
如果数组长度为n,窗口大小为w,则一共产生n-w+1个窗口的最大值。
请实现一个函数。
- 输入:整型数组arr,窗口大小为w。
- 输出:一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。
代码实现
public class GetMaxWindows {
public int[] getMaxWindows(int[] arr, int w) {
if (arr == null || w < 1 || arr.length < w) {
return null;
}
LinkedList<Integer> qmax = new LinkedList<>(); //存放数组arr中的下标
int[] res = new int[arr.length - w + 1];
int index = 0;
for (int i = 0; i < arr.length; i++) {
while (!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]) {
qmax.pollLast();
}
qmax.addLast(i);
if (qmax.peekFirst() == i - w) { //当前qmax队头是否过期
qmax.pollFirst();
}
if (i >= w - 1) {
res[index++] = arr[qmax.peekFirst()];
}
}
return res;
}
public static void main(String[] args) {
GetMaxWindows gmw = new GetMaxWindows();
int[] arr = new int[]{4, 3, 5, 4, 3, 3, 6, 7};
System.out.println(Arrays.toString(gmw.getMaxWindows(arr, 3)));
}
}