騰訊三面算法題-逛街（單調棧）

原問題： 看到多少棟樓？

原題鏈接

小Q在週末的時候和他的小夥伴來到大城市逛街，一條步行街上有很多高樓，共有n座高樓排成一行。
小Q從第一棟一直走到了最後一棟，小Q從來都沒有見到這麼多的樓，所以他想知道他在每棟樓的位置處能看到多少棟樓呢？（當前面的樓的高度大於等於後面的樓時，後面的樓將被擋住）

解法：用單調棧

package com.company;
import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int[] arr = new int[n];
        int[] ans = new  int[n];

        for(int i = 0; i < n; i++){
            arr[i]=in.nextInt();
        }
        Stack<Integer> L = new Stack<>();
        for (int i = 0; i < n;i++){
            ans[i] = L.size();//保存的是i往左看，能看到的全部樓高和
            if (i != 0) {
                while (!L.empty() && arr[i] >= L.peek()) {//如果這個樓能把它前面的樓擋住，那就把被擋住的樓移除
                    L.pop();
                }
            }
            L.push(arr[i]);
        }
        Stack<Integer> R = new Stack<>();
        for (int i = n-1; i >= 0;i--){
            ans[i] += R.size();//保存的是i往右看，能看到的全部樓高
            if (i != n-1) {
                while (!R.empty() && R.peek() <= arr[i]) {
                    R.pop();
                }
            }
            R.push(arr[i]);
        }
        for (int i = 0; i < n; i++){
            System.out.print((ans[i]+1)+" ");
        }
    }

}

類似問題：LeetCode每日溫度

leetcode 739. Daily Temperatures

1. 問題化簡:找後面第一個比他大的元素。
2. 首先想到從前往後遍歷，這樣會導致有大量重複計算的。
3. 然後想到從後往前遍歷，保存一個從上往下遞增的棧，如果新來的元素比前面的元素比棧頂元素大的話，就會破壞棧的單調性，所以出棧，直到當前元素比棧頂元素大。
4. 實現細節一：如果棧內保存的是溫度，統計出棧次數作爲間隔天數。這樣這個棧只能統計出一次的結果。比如說 計算75 71 69 72。我們是從後往前遍歷。求71的時候,棧裏是69 72這時候把69出棧了，算出71的天數是2。那求75的時候怎麼辦？69已經沒了啊！
5. 所以棧裏應該保存index，計算index之差作爲間隔天數。
6. 實現細節二：如果後面沒有比他更高的溫度，輸出0。怎麼判斷呢？ 好辦，因爲每次棧頂元素>=當前元素時（注意這個等號），出棧，如果棧都空了，自然就是沒找到。
7. 實現細節三：題幹是找出更溫暖的天，那溫度<=當前溫度的天都要被出棧，所以別忘了加等號。

package com.company;

import java.util.Stack;

class Solution {
    public int[] dailyTemperatures(int[] T) {
        int len = T.length;
        int ans[] = new int[len];
        Stack<Integer> stack = new Stack<>();
        if (len <= 1){
            return ans;
        }
        stack.push(len-1);
        for (int i = len-2; i >= 0;i--){
            while (!stack.empty() && T[i] > T[stack.peek()]){
                stack.pop();
            }
            if (stack.empty()){
                ans[i] = 0;
            }else {
                ans[i] = stack.peek()-i;
            }
            stack.push(i);
        }
        return ans;
    }
}