GuilinDev

Lc1944

05 August 2008

1944. Number of Visible People in a Queue

队列中可以看到的人数

有 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 个人在他右侧队列中能 看到 的 人数 。

示例:

输入: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 个人谁也看不到因为他右边没人。

思路

单调栈。栈保存下标。从右向左扫描,每一次弹出 比当前元素小的右边元素,结果 +1。如果弹出之后,栈还是非空,那结果还得 +1。让自身入栈。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
    public int[] canSeePersonsCount(int[] heights) {
        int len = heights.length;
        int[] result = new int[len];

        Deque<Integer> stack = new ArrayDeque<>();
        stack.push(len - 1);

        for (int i = len - 2; i >= 0; i--) {
            while (!stack.isEmpty() && heights[i] > heights[stack.peek()]) {
                stack.pop();
                result[i]++;
            }
            if (!stack.isEmpty()) {
                result[i]++;
            }
            stack.push(i);
        }

        return result;
    }
}