当前位置:   article > 正文

二刷力扣——单调栈

二刷力扣——单调栈

739. 每日温度

单调栈应该从栈底到栈顶 是递减的。

找下一个更大的 ,用递减单调栈,就可以确定在栈里面的每个比当前元素i小的元素,下一个更大的就是这个i,然后弹出并记录;然后当前元素i入栈,仍然满足递减要求。最后留在栈里面的是没有下一个更大的值的,符合要求。

找下一个更小的用递增单调栈同理。

result[被弹出下标]=使他弹出下标-被弹出下标。

复习一下Java集合知识:来自Java集合详解-CSDN博客

LinkedList 可以使用多种接口进行声明,包括但不限于:

  • Deque 接口:Deque<Integer> deque = new LinkedList<>();: 双端队列的操作,包括栈和队列的功能。
  • Queue 接口:Queue<Integer> queue = new LinkedList<>();: 适合实现队列的操作,包括 offer、poll、peek 等方法。
  • List 接口:List<Integer> list = new LinkedList<>();: 通用的集合操作,如添加、删除、获取元素等。
  • Collection 接口:Collection<Integer> collection = new LinkedList<>();: 这种声明方式表示通用的集合操作,适合需要简单地操作元素集合的场景。
  1. class Solution {
  2. public int[] dailyTemperatures(int[] temperatures) {
  3. int n=temperatures.length;
  4. Deque<Integer> stack=new LinkedList<>();//放下标,
  5. int [] result =new int[n];
  6. for(int i=0;i<n;++i)
  7. {
  8. int temp=temperatures[i];
  9. while(!stack.isEmpty() && temperatures[stack.peek()]<temp)
  10. {
  11. int a=stack.pop();//小,被弹出
  12. result[a]=i-a;
  13. }
  14. //找到位置
  15. stack.push(i);
  16. }
  17. return result;
  18. }
  19. }

时间、空间O(n)

496. 下一个更大元素 I

跟上一题区别是,先用单调栈把nums2处理,较小的元素弹出来的时候要找到当前元素i在nums1的位置,然后给到result。

这里不算下标而且数组里面没有重复元素,所以单调栈里面可以放元素值。

另外result一开始都赋值-1。最后可能nums1里有几个是没有下一个更大值的。

  1. class Solution {
  2. public int[] nextGreaterElement(int[] nums1, int[] nums2) {
  3. int m=nums1.length ,n=nums2.length ;
  4. Deque<Integer> stack=new LinkedList<Integer>();
  5. int[] result=new int[m];
  6. HashMap<Integer,Integer> hash=new HashMap<>();// 放元素
  7. for(int i=0;i<m;++i)
  8. {
  9. hash.put(nums1[i],i);
  10. }
  11. for(int i=0;i<m ;++i)
  12. {
  13. result[i]=-1;
  14. }
  15. for(int i=0;i<n;++i)
  16. {
  17. int tmp=nums2[i];
  18. while(!stack.isEmpty() && stack.peek()<tmp)
  19. {
  20. int a=stack.pop();//被弹出的元素a
  21. int pos=hash.getOrDefault(a,-1);
  22. if(pos!=-1)result[pos]=tmp;
  23. }
  24. stack.push(tmp);
  25. }
  26. return result;
  27. }
  28. }

时间是O(m+n)。初始化result和hash的是O(m),单调栈循环是O(n),因为用的HashMap,查找O(1)。

503. 下一个更大元素 II

要求循环地搜索元素的下一个更大的数,其实就是单调栈要走两次nums数组。

  1. class Solution {
  2. public int[] nextGreaterElements(int[] nums) {
  3. int n=nums.length;
  4. int[] result=new int[n];
  5. Deque<Integer> stack=new LinkedList<>();
  6. Arrays.fill(result,-1);
  7. for(int j=0;j<2*n;++j)
  8. {
  9. int i=j%n;
  10. int num=nums[i];
  11. while(!stack.isEmpty() && nums[stack.peek()]<num)
  12. {
  13. int a=stack.pop();
  14. result[a]=nums[i];
  15. }
  16. stack.push(i);
  17. }
  18. return result;
  19. }
  20. }

42. 接雨水

1、以列为单位,每列都找自己左边(包含自己)的最高柱子 l 和自己右边(包含自己)的最高柱子 r,就形成一个凹槽。然后每一列应该装下的雨水数就应该是(min(l , r) - 自己柱子高)*宽。所以主要关注l 和 r 怎么计算。

1.1、DP

维护一个lefts和rights数组,lefts[i]是l,rights[i]就是r。

那么递推公式就是:(min(lefts[i] , rights[i]) - 自己柱子高)*宽

关键得到这两个数组,还是很直观的,lefts涉及到的是i及左边的,所以从左往右递推;rights从右往左。

  1. class Solution {
  2. public int trap(int[] height) {
  3. int n=height.length;
  4. int count=0;
  5. int[] lefts=new int[n];
  6. int[] rights=new int[n];
  7. lefts[0]=height[0];
  8. for(int i=1;i<n;++i)
  9. {
  10. lefts[i]=Math.max(lefts[i-1],height[i]);
  11. }
  12. rights[n-1]=height[n-1];
  13. for(int i=n-2;i>=0;--i)
  14. {
  15. rights[i]=Math.max(rights[i+1],height[i]);
  16. }
  17. for(int i=0;i<n;++i)
  18. {
  19. count+=(Math.min(lefts[i],rights[i])-height[i]);
  20. }
  21. return count;
  22. }
  23. }

时间空间:O(n)

1.2、双指针

用两个指针left、right从两端,逐渐逼近,逼近的判断条件是:left指向的和right指向的对比,哪一方小,就走一步。

①这里对于红框的判断不能理解,为什么当前的哪一方小,就能确定这一方的最大值更小呢?

下面这个说法很形象。其实就是A队B队比赛,遍历过了的都是输了的(或者平局的),假如B队的curB赢了A队的curA,所以目前MAX_B(其实就是curB)是目前两队最强。即A队最强的其实是输给B队过的了,所以A队最强不如B队最强。可以确定MAX_B=curB>MAX_A。A赢B就同理。

②又想:假如B队赢了A队,可以确定出现了一个凹槽:A队最高、curA、B队最高。这个凹槽的两端一定是我们要求的吗?

A队最高肯定是所要求的左边最高柱子l ,B队最高不一定是右边最高柱子 r。假如curA和curB之间还有比MAX_A更矮的柱子,压根不考虑因为我们是先找两端最高的,再在两个最高的里找最小的;假如有比MAX_B更高的柱子,那取Min值仍然还是MAX_A。A赢B就同理。

总的来说,哪一方输了,哪一方的输家就可以确定他的雨水了。决定这个雨水高度的就是他这一方的最强者(但是比另一方的最强者弱)。

可以用MAX_B>MAX_A,用height[a]<height[b]感觉更能体现这个“比赛”的过程吧,当前选手的大小。

  1. class Solution {
  2. public int trap(int[] height) {
  3. int n=height.length;
  4. int a=0,b=n-1;//对手
  5. int MAX_A=0,MAX_B=0;//一个队里面已比对手的最强者
  6. int count=0;
  7. while(a<b)
  8. {//更新最强者,要包括当前选手
  9. MAX_A=Math.max(MAX_A,height[a]);
  10. MAX_B=Math.max(MAX_B,height[b]);
  11. int yushui=0;//输的一方收集雨水
  12. if(height[a]<height[b])//b整体赢了
  13. {
  14. yushui=MAX_A-height[a];//注意凹槽两端
  15. a++;//输的下场,派下一个
  16. }
  17. else if(height[a]>=height[b])
  18. {
  19. yushui=MAX_B-height[b];
  20. b--;
  21. }
  22. count+=yushui;
  23. }
  24. return count;
  25. }
  26. }

时间O(n)空间O(1)

2、以行为单位

2.1 单调栈

以列为单位求,每个柱子的雨水是一次性求好,需要提前知道左边最高柱子和右边最高柱子,这形成凹槽1;

以行为单位求,是求以每个柱子为底的这一行能接的雨水,这一行雨水可以到的高度,应该是最接近自己的柱子的最小高度。即Min(上一个更大,下一个更大)。这是凹槽2。

这是按行求 和 按列求 的主要区别。

这一行,宽是 下一个更大和上一个更大的 下标间隔。高是Min值和中间 a 的差。

  1. class Solution {
  2. public int trap(int[] height) {
  3. int n=height.length;
  4. int count=0;
  5. Deque<Integer> stack=new LinkedList<Integer> ();
  6. for(int i=0;i<n;++i)
  7. {
  8. int tmp=height[i];
  9. while(!stack.isEmpty() && height[stack.peek()]<tmp)//凹槽
  10. {
  11. int a=stack.pop();//l=栈底,a,r=height[i]
  12. if(!stack.isEmpty()){
  13. int l=stack.peek();
  14. int r=i;
  15. int kuan=r-l-1;
  16. int gao=Math.min(height[l],height[r])-height[a];
  17. count+=kuan*gao;
  18. }
  19. }
  20. stack.push(i);
  21. }
  22. return count;
  23. }
  24. }

84. 柱状图中最大的矩形

貌似只能以行为单位,所以只看单调栈方法。

需要找到元素i的上一个更小的元素leftmin和下一个更小的元素rightmin,这样leftmin和rightmin之间的元素都比当前元素i更大,那么矩形的宽就是中间的这些元素:可以从leftmin+1延伸到rightmin-1,长即为height[i]。

怎么确定矩形的宽 和高?宽有左边界和右边界,这里感觉应该每次固定一边然后讨论另一边。这里固定右边界,是遍历的i,也就是对于每个柱子i 都计算,以这个元素为右边界 ,矩形的最大面积。知道右边界了,要让面积最大左边界应该是多少呢?其实应该找上一个更小值l 。这样 (l,i) 之间都是比i 高或者相等的柱子,这就是固定i 为右边界的最大面积。

怎么保证能处理所有元素?使用单调栈,pop一个元素就算这个元素右边界的矩形。题目说柱子都是>=0的,那么在最后的时候 ,单调栈再插入一个0,就能把栈里面的所有元素都pop出来,就都处理到了

单调栈 的顺序 应该是 需要 严格单调递增的。因为栈不为空要找上一个更小值l 。之前的题事实上只在单个>、<的时候要出栈,因为只需要求下一个更大/更小。上面的接雨水也是栈顶=当前的话没关系,相减是0 没影响。但是这里得严格递增。

过程是:当 i 元素 <= 栈顶的 a,这个时候这个i 元素就是右边界,我们要讨论单调栈里面所有比这个元素大的元素为高的矩形面积。现在 两种情况:

1、栈顶a弹出,栈为空了,说明什么?在heights里面,a的左边没有比他严格小于的l ,都是>=他的,而a 的下一个更小值又是i,所以区间是[0,i-1)所以宽就是i;

2、如果栈不为空,那就是存在l ,这个l 就是a弹出之后的新栈顶(比a 要小的)。所以矩形区间是(新栈顶=l,i)。

  1. class Solution {
  2. public int largestRectangleArea(int[] heights) {
  3. int n = heights.length;
  4. Deque<Integer> stack = new LinkedList<>(); // 存储柱子的索引
  5. int maxArea = 0;
  6. for (int i = 0; i <= n; ++i) {
  7. int h = (i == n) ? 0 : heights[i]; // 最后加入一个高度为0的柱子,确保处理完所有柱子
  8. while (!stack.isEmpty() && heights[stack.peek()] >= h) {
  9. int a=stack.pop();
  10. int height = heights[a]; // 当前出栈的柱子高度
  11. // 计算宽度:是i代表前面i个都是>=height的。否则矩形宽是(上一个更小的,i-1]
  12. int width = stack.isEmpty() ? i : i - stack.peek() - 1;
  13. maxArea = Math.max(maxArea, height * width);
  14. System.out.println(maxArea);
  15. }
  16. stack.push(i); // 当前柱子的索引入栈
  17. }
  18. return maxArea;
  19. }
  20. }

主要是 :确定一边求另一边的思想、求上一个更小值用单调栈、为了处理所有元素添加一个最小值0。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/一键难忘520/article/detail/796995
推荐阅读
相关标签
  

闽ICP备14008679号