有 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 <= 10^5
1 <= heights[i] <= 10^5
heights 中所有数 互不相同 。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/number-of-visible-people-in-a-queue 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
vector<int> canSeePersonsCount(vector<int>& heights) {
int n = heights.size();
vector<int> ans(n, 0);
stack<int> s;
for(int i = n-1; i >= 0; i--)
{
bool flag = false;//是否见过比自己矮的
while(!s.empty() && heights[i] > heights[s.top()])
{
s.pop();//后续的人肯定看不见
flag = true;
ans[i]++;//见过一个比自己矮的,+1,可以看见
}
if(!flag && i != n-1)
ans[i] = 1;//除了最右边的人,至少可以看见1个人
else if(flag && !s.empty())
ans[i]++;//如果见过比自己矮的人,且栈内还有人(比自己高,能看见)
s.push(i);
}
return ans;
}
};
144 ms 82.1 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!