數組滑動窗口比較最大值

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Deque;
import java.util.List;
import java.util.concurrent.LinkedBlockingDeque;

/**
 * 有一個×××數組arr和一個大小爲w的窗口從數組的最左邊滑到最右邊,窗口每次向右滑動一個位置。
 * 返回一個長度爲n-w+1的數組res,res[i]表示每一種窗口狀態下的最大值。
 * 
 * 例:
 * 數組arr=[4,3,5,4,3,3,6,7] w=3,結果爲[5,5,5,4,6,7]
 * 
 * 常規解連續比較w內的最大數,O(N*w)
 * 
 * 最優解O(N)
 * @author Administrator
 *
 */
public class CompareArray {
	/**
	 * 雙端隊列存放着數組的下標值
	 */
	private static Deque<Integer> qmax = new LinkedBlockingDeque<Integer>();
	
	public static List dowork(int[] arr,int size) {
		List<Integer> l = new ArrayList<Integer>();
		for(int i = 0,len = arr.length;i<len;i++) {
			
			//1、如果qmax爲空,直接把下標i放入qmax
			if(qmax.isEmpty()) {
				qmax.push(i);
			}else {
				//2、如果qmax末尾<=arr[i],則彈出
				while(!qmax.isEmpty() && arr[qmax.getLast()]<=arr[i]) {
					qmax.removeLast();
				}
				qmax.push(i);
			}
			
			//超過窗口大小的彈出
			if(i - qmax.peekLast() == size) {
				qmax.removeLast();
			}
			
			//拿到窗口內最大值
			if(i>=size-1) {
				l.add(arr[qmax.peekLast()]);
			}
		}
		
		return l;
	}
	
	public static void main(String[] args) {
		int[] i = {4,3,5,4,3,3,6,7};
		
		System.out.println(dowork(i,5));
		
	}
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章