LeetCode-1994-队列中可以看到的人

LeetCode-1994-队列中可以看到的人

有 n 个人排成一个队列，从左到右 编号为 0 到 n - 1 。给你以一个整数数组 heights ，每个整数 互不相同，heights[i] 表示第 i 个人的高度。

一个人能 看到 他右边另一个人的条件是这两人之间的所有人都比他们两人 矮 。更正式的，第 i 个人能看到第 j 个人的条件是 i < j 且 min(heights[i], heights[j]) > max(heights[i+1], heights[i+2], ..., heights[j-1]) 。

请你返回一个长度为 n 的数组 answer ，其中 answer[i] 是第 i 个人在他右侧队列中能 看到 的 人数 。

示例 1：

输入：heights = [10,6,8,5,11,9]
输出：[3,1,2,1,1,0]
解释：
第 0 个人能看到编号为 1 ，2 和 4 的人。
第 1 个人能看到编号为 2 的人。
第 2 个人能看到编号为 3 和 4 的人。
第 3 个人能看到编号为 4 的人。
第 4 个人能看到编号为 5 的人。
第 5 个人谁也看不到因为他右边没人。

示例 2：

输入：heights = [5,1,2,3,10]
输出：[4,1,1,1,0]

提示：

• n == heights.length
• 1 <= n <= 105
• 1 <= heights[i] <= 105
• heights 中所有数 互不相同 。

class Solution {
    /**
     * 单调栈, 单调递减
     * 从右往左遍历, 如果栈中有数据并且当前数字比栈顶数据大, 栈顶元素出栈, 继续判断直至小于栈顶元素或栈为空,
     * 这一过程中, 出栈元素的个数 + (栈是否不为空 ? 1 : 0) 即是该元素能够看到的数据个数
     */
    public int[] canSeePersonsCount(int[] heights) {
        Deque<Integer> stack = new ArrayDeque<>();
        int size = heights.length;
        int[] counts = new int[size];
        for (int i = size - 1; i >= 0; i--) {
            int cnt = 0;
            while (!stack.isEmpty() && heights[i] > stack.peek()) {
                stack.pop();
                cnt++;
            }
            counts[i] = cnt + (stack.isEmpty() ? 0 : 1);
            stack.push(heights[i]);
        }
        return counts;
    }
}