【Lintcode】285. Tall Building

题目地址：

https://www.lintcode.com/problem/tall-building/description

给定一个数组$A$，代表一列大楼的高度。问站在每个位置$i$向左右两边看，能看到的大楼的数量是多少。这个数量首先包含$A[i]$这个楼本身，其次从这个位置向左右两边看的时候，高的楼会把后面高度相等或者更低的楼挡住。最后返回一个数组即可。

思路是单调栈。我们考虑一下站在$i$这个位置，能看到的大楼都有哪些。例如向左看，那么$A[i-1]$是能看到的，接着先从左到右遍历数组$A$，同时开一个严格下降的单调栈，这个栈维护的是从当前位置向左看能看到的大楼的高度。所以栈的size就是向左看能看到的大楼的数量。同理，再从右向左遍历一下数组$A$，就得到了从某个位置向右看能看到的大楼的高度。加起来就是能看到的大楼总数。代码如下：

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;

public class Solution {
    /**
     * @param arr: the height of all buildings
     * @return: how many buildings can he see at the location of each building
     */
    public int[] tallBuilding(int[] arr) {
        // Write your code here.
        int[] res = new int[arr.length];
        Arrays.fill(res, 1);
        
        Deque<Integer> stack = new ArrayDeque<>();
        for (int i = 0; i < arr.length; i++) {
            res[i] += stack.size();
            while (!stack.isEmpty() && stack.peek() <= arr[i]) {
                stack.pop();
            }
            
            stack.push(arr[i]);
        }
        
        // 清空栈，再从右向左遍历数组
        stack.clear();
    
        for (int i = arr.length - 1; i >= 0; i--) {
            res[i] += stack.size();
            while (!stack.isEmpty() && stack.peek() <= arr[i]) {
                stack.pop();
            }
    
            stack.push(arr[i]);
        }
        
        return res;
    }
}

时空复杂度$O(n)$。