Java算法真題詳解運用單調棧

更新時間：2022年07月21日 11:40:45 作者：風鈴聽雨~

一般使用單調棧無非兩個方向，單調遞減，單調遞增。單調遞增棧：存進去的數(shù)據(jù)都是增加的，碰到減少的時候，這時就要進行操作了。單調遞減棧：存進去的數(shù)據(jù)都是減少的，碰到增加的時候，這時就要進行操作了，下面我們在真題中運用它

1.下一個更大元素

題目描述

思路詳解

這一題就選取比較暴力的解法了。

我們先初始化一個與nums等長度的res數(shù)組用來存儲結果，我們遍歷取出nums中的值，到nums2中尋找，直到找到nums2[j] == nums[i] ,我們再從 nums2的 j 之后遍歷找到比nums[i]大的數(shù)組返回，找不到就返回-1.

代碼與結果

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int m = nums1.length, n = nums2.length;
        int[] res = new int[m];
        for (int i = 0; i < m; ++i) {
            int j = 0;
            while (j < n && nums2[j] != nums1[i]) {
                ++j;
            }
            int k = j + 1;
            while (k < n && nums2[k] < nums2[j]) {
                ++k;
            }
            res[i] = k < n ? nums2[k] : -1;
        }
        return res;
    }
}

2.每日溫度

題目描述

思路詳解

這一題也是使用比較暴力的方法。同上一題一樣。

兩重循環(huán)，顯然這種方法時間復雜度較高。這里也提供一種時間復雜度較低的方法。

單調棧

我們維護的是一個差距數(shù)組也就是結果數(shù)組，首先創(chuàng)建一個棧，判斷棧是否為空，若為空直接入棧，若不為空則比較棧頂元素與當前元素，如果當前元素大于當前元素就把差值放入到對應結果數(shù)組同時棧頂元素出棧，若當前元素小于結果數(shù)組則入棧。

這里給一個動畫鏈接很好董，動畫學單調棧。

代碼與結果

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int[] answer = new int[temperatures.length];
        for(int i = 0 ; i < temperatures.length ; i++){
            int res = 0;
            for(int j = i+1 ; j < temperatures.length; j++){
                res++;
                if(temperatures[j] > temperatures[i]){
                    answer[i] = res;
                    break;
                } 
            }
        }
        return answer;
    }
}

3.下一個更大元素II

題目描述

思路詳解

本題的思路呢單調棧，問題一暴力破解，這個題要使用單調棧了。

原理同第二題的方法一樣，就在循環(huán)上注意一下就可以了。直接上代碼，不董的可以看第二題的視頻再來看這個哦。

代碼與結果

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] ret = new int[n];
        Arrays.fill(ret, -1);
        Deque<Integer> stack = new LinkedList<Integer>();
        for (int i = 0; i < n * 2 - 1; i++) {//最多循環(huán)一次，只需要2*n-1就夠用了
            while (!stack.isEmpty() && nums[stack.peek()] < nums[i % n]) {
                ret[stack.pop()] = nums[i % n];
            }
            stack.push(i % n);//利用取模來進行循環(huán)也是一種常用的方法
        }
        return ret;
    }
}