腾讯三面算法题-逛街（单调栈）

原问题： 看到多少栋楼？

原题链接

小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;
    }
}