问题
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。
例如,数组为[4,3,1,5,4,3,7,5],窗口大小为5时:
[4 3 1 5 4] 3 7 5 max = 5
4 [3 1 5 4 3] 7 5 max = 5
4 3 [1 5 4 3 7] 5 max = 7
4 3 1 [5 4 3 7 5] max = 7
即窗口最大值数组为 result = {5, 5,7,7}
解一:暴力解法
import java.util.Arrays;
class Main{
public static void main(String[] args) {
int[] arrr = {4,3,1,5,4,3,7,5};
int wr = 5;
getMaxWindow(arrr,wr);
}
public static int[] getMaxWindow(int [] arr,int w){ //w是窗口的大小
if(w<1 || arr==null || arr.length<w){
return null;
}
int[] result = new int[arr.length-w+1];
for(int i= 0;i<arr.length-w+1;i++){
int max = arr[i];
for(int j = i;j<i+w;j++){
if(arr[j]>max) {
max = arr[j];
}
}result[i] = max;
}
// for(int i =0;i<result.length;i++){
// System.out.println(arr[i]);
// }
// System.out.println(result); //上面两种打印方法都不行
System.out.println(Arrays.toString(result));
return result;
}
}
突然知道java打印一维数组的方式是 System.out.println(Arrays.toString(result));
解二:使用双向队列结构
优化 使用双向队列结构,,每次更新左边一定不会再出现最大值,右边可能会出现最大值,那就把右边单独做成双向队列结构,队列底放最大值,出现最大的值 队列内前面的数字就清空,把最大的数放在最下面
import java.util.Arrays;
import java.util.LinkedList;
class Main{
public static void main(String[] args) {
int[] arrr = {4,3,1,5,4,3,7,5};
int wr = 5;
getMaxWindow(arrr,wr);
}
public static int[] getMaxWindow(int[] arr,int w){
if (w < 1 || arr == null || arr.length < w) {
return null;
}
int index =0;
int[] result = new int[arr.length - w + 1];
LinkedList<Integer> temp = new LinkedList<>();
for(int i=0;i<arr.length;i++){//队列不为空而且这个数字比队列首要大的话
while (!temp.isEmpty() &&arr[temp.peekLast()]<=arr[i]){
temp.pollLast(); // 把队尾元素弹出
}
temp.addLast(i); //把这个元素放到队尾
if(temp.peekFirst() == i -w){
temp.pollFirst(); //如果队首元素不再窗口范围内,则弹出
}
//经过上面的,放在队列首端的就一定是最大的,把他副那个在result数组中
if(i>=w-1){
result[index++] = arr[temp.peekFirst()];
}
}
System.out.println(Arrays.toString(result));
return result;
}
}