棧和隊列——最大值減去最小值小於或等於num的子數組數量

【題目】
  給定數組arr和整數num,共返回有多少個子數組滿足如下情況:
  max(arr[i…j]) - max(arr[i…j]) <= num
  max(arr[i…j])表示子數組arr[i…j]中的最大值,min(arr[i…j])表示子數組arr[i…j])中的最小值

  普通的邏輯解法就是,找到arr的所有子數組,然後對每一個子數組做遍歷找到其中的最小值和最大值,然後看看這個子數組是否滿足條件,統計所有滿足的子數組數量即可。這種邏輯比較簡單,這裏不做解釋,下面的做法實現時間複雜度和空間複雜度都是O(N)

【代碼實現】

import java.util.*;

class test{
    public static int getNum(int[] arr,int num){
        if(arr==null || arr.length==0){
            return 0;
        }
        LinkedList<Integer> qmin = new LinkedList<Integer>();
        LinkedList<Integer> qmax = new LinkedList<Integer>();
        int i = 0;
        int j = 0;
        int res = 0;
        while(i<arr.length){
            while(j<arr.length){
                while(!qmin.isEmpty() && arr[qmin.peekLast()]>=arr[j]){
                    qmin.pollLast();
                }
                qmin.addLast(j);
                while(!qmax.isEmpty() && arr[qmax.peekLast()]<=arr[j]){
                    qmax.pollLast();
                }
                qmax.addLast(j);
                if(arr[qmax.peekFirst()]-arr[qmin.peekFirst()]>num){
                    break;
                }
                j++;
            }
            if(qmin.peekFirst()==i){
                qmin.pollFirst();
            }
            if(qmax.peekFirst()==i){
                qmax.pollFirst();
            }
            res += j - i;aaaaaaaaaaa
            i++;
        }
        return res;
    }

    public static void main(String[] args){
        int[] arr = new int[]{3,10,4,13,16,7};
        int res = getNum(arr,7);
        System.out.println(res);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章