2024.1.5每日一题

LeetCode每日一题

1944.队列中可以看到的人数

1944. 队列中可以看到的人数 - 力扣(LeetCode)

题目描述

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

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

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

示例 1:

2024.1.5每日一题_第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 中所有数 互不相同

思路

单调栈

代码

C++
class Solution {
public:
    vector canSeePersonsCount(vector& heights) {
        int n = heights.size();
        vector stack;
        vector res(n,0);

        for(int i = n - 1; i >= 0; i--){
            int h = heights[i];

            while(!stack.empty() && stack.back() < h){
                stack.pop_back();
                res[i] += 1;
            }
            if(!stack.empty()){
                res[i] += 1;
            }
            stack.push_back(h);
        }
        return res;
    }
};
Java
class Solution {
    public int[] canSeePersonsCount(int[] heights) {
        int n = heights.length;
        Deque<Integer> stack = new ArrayDeque<Integer>();

        int[] res = new int[n];

        for(int i = n - 1; i >= 0; i--){
            int h = heights[i];
            while(!stack.isEmpty() && stack.peek() < h){
                stack.pop();
                res[i] += 1;
            }
            if(!stack.isEmpty()){
                res[i]++;
            }
            stack.push(h);
        }
        return res;
    }
}

2024.1.5每日一题_第2张图片

你可能感兴趣的:(算法学习,#,每日一题,算法,leetcode)