LeetCode1944. 队列中可以看到的人数-----单调栈

题目表述

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

单调栈

逆序遍历数组,用递增栈来存储身高(从栈顶->栈底升序)。
逆序遍历这 n 个人的下标,如果当前遍历到了第 i 个人,那么我们需要在栈中选出第 i 个人可以看到的那些人。设栈顶的下标为stack.peek(),则:

  • 如果栈为空,说明第 i 个人是遍历到的所有人中最高的那个人,退出比较环节;
  • 如果height[i] > height[stack.peek()],说明i能够看到栈顶元素,并且i左侧的人均看不见,因此将栈顶弹出,并与新的栈顶比较。最后判断栈是否为空,非空的话能看到元素+1;
  • 如果height[i] < height[stack.peek()],说明i能够看到栈顶,但是看不到右边的人,所以退出比较环节。
class Solution { public int[] canSeePersonsCount(int[] heights) { Deque<Integer> stack = new ArrayDeque<>(); int[] res = new int[heights.length]; for(int i = heights.length - 1; i >= 0;i--){ while(!stack.isEmpty() && heights[i] > heights[stack.peek()]){ res[i] += 1; stack.pop(); } if(!stack.isEmpty()) res[i] += 1; stack.push(i); } return res; } }

__EOF__

本文作者Younger
本文链接https://www.cnblogs.com/youngerwb/p/16259605.html
关于博主:评论和私信会在第一时间回复。或者直接私信我。
版权声明:本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!
声援博主:如果您觉得文章对您有帮助,可以点击文章右下角推荐一下。您的鼓励是博主的最大动力!
posted @   YoungerWb  阅读(39)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示