代码随想录打卡—day59—【单调栈】— 9.5 单调栈2

1 503. 下一个更大元素 II

503. 下一个更大元素 II

for俩遍+vis访问数组+单调栈

AC：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        //for*2单调栈
        vector<int> ans(nums.size(),-1);
        stack<int>sk;
        vector<bool> vis(nums.size(),0);
        for(int i = 0,k = 0; k < 2*nums.size();k++,i++,i %= nums.size())
        {
            if(sk.empty() || nums[i] <= nums[sk.top()])
            {
                if(vis[i] == 0)
                    sk.push(i);
            }
            else
            {
                while(!sk.empty() && nums[i] > nums[sk.top()] )
                {
                    ans[sk.top()] = nums[i];
                    vis[sk.top()] = 1;
                    sk.pop();
                }
                if(vis[i] == 0)
                    sk.push(i);
            }
        }
        return ans;
    }
};

看了题解：应该不加vis 直接重复加之前访问过的元素 也没事，原因是它还是值会被右边第一个大于自身的元素弹出 就是重复操作一次。确实，AC：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        //for*2单调栈
        vector<int> ans(nums.size(),-1);
        stack<int>sk;
        for(int i = 0,k = 0; k < 2*nums.size();k++,i++,i %= nums.size())
        {
            if(sk.empty() || nums[i] <= nums[sk.top()])
            {
                sk.push(i);
            }
            else
            {
                while(!sk.empty() && nums[i] > nums[sk.top()] )
                {
                    ans[sk.top()] = nums[i];
                    sk.pop();
                }
                sk.push(i);
            }
        }
        return ans;
    }
};

2 42. 接雨水

42. 接雨水

【解法1：】自己写的双指针 就是逻辑不清晰，但也过了

一开始的思路： lr一开始先找到非0元，然后r找到第一个>= height[l]的元素 ans+= ，l = r 重复或者r到头了结束 。但是这样不能处理样例。于是又想到 右边height.size()-1 至0 右边往左边重复一次，取两次最大值。 过得了样例 但是还是不对。处理不了 [6,4,2,0,3,2,0,3,1,4,5,3,2,7,5,3,0,1,2,1,3,4,6,8,1,3] 类似的样例

最终思路：所以还是要左到右一次过 ，并且在最后一个高点找不到右匹配时候，右往左（左边界为新的高点l）再来一次。

AC代码：不可思议，完全一个脑子想出ac的第一道困难题，泰酷辣！

class Solution {
public:
    int trap(vector<int>& height) 
    {
        //双指针
        int l = 0;
        int r = 0;
        // 思路： lr一开始先找到非0元，然后r找到第一个>= height[l]的元素 ans+= ，l = r 重复或者r到头了结束 
        //  r到头了还是存在没一段没处理的部分，所以所以需要以当前的l作为左边界 右边往左边再来一次
        while(l < height.size() && height[l] == 0 )l++;
        r = l+1;
        int ans1 = 0;
        int tmp1 = 0;
        while(l < height.size())
        {
            while( r < height.size() && height[r] < height[l])
            {
                tmp1 += (height[l] - height[r]);
                r++;
            }
            if(r < height.size())
            {
                ans1 += tmp1;
                tmp1 = 0;
                l = r;
                r++;
            }

            if(r == height.size())   // 开始反向走
            {
                int end_l = l;
                l = height.size() - 1;
                r = l - 1;
                tmp1 = 0;

                while(r >= end_l )
                {
                    while(r >= end_l && height[r] < height[l])
                    {
                        tmp1 += (height[l] - height[r]);
                        r--;
                    }
                    if(r >= end_l)
                    {
                        ans1 += tmp1;
                        tmp1 = 0;
                        l = r;
                        r--;
                    }
                }

                break;
            }
        }
        cout << ans1 << endl;


        return ans1;
    }
};

【解法2：】双指针第2版-->左右两个数组大法

AC：

class Solution {
public:
    int trap(vector<int>& height) 
    {
        //双指针第2版-->左右两个数组大法
        /*
            关键：
            把问题转换成
            每个格子应该的补充的高 是其左侧最高和右侧最高的min-当前高
        */
        vector<int> left(height.size());  // 元素i左侧最大的元素
        vector<int> right(height.size());  //元素i右侧最大的元素

        int left_max = 0;
        int right_max = 0;

        for(int i = 0; i < height.size();i++)
        {
            if(i != 0)left[i] = left_max;
            left_max = max(left_max,height[i]);
        }
        for(int i = height.size()-1; i >= 0;i--)
        {
            if(i != height.size()-1)right[i] = right_max;
            right_max = max(right_max,height[i]);
        }      
        int ans = 0;
        for(int i = 0; i < height.size();i++)
        {
            if(i == 0 || i == height.size())continue;
            if(min(left[i],right[i]) > height[i])
                ans += (min(left[i],right[i])-height[i]);
        }

        return ans;
    }
};

【解法3：】单调栈做法

AC:

class Solution {
public:
    int trap(vector<int>& height) 
    {
        // 单调栈
        stack<int> sk;
        int ans = 0;
        sk.push(0); // 懒得多一个判断栈空的if
        for(int i = 1; i < height.size();i++)
        {
            if(height[i] < height[sk.top()])
            {
                sk.push(i);
            }
            else if(height[i] == height[sk.top()])
            {
                sk.pop();
                sk.push(i);
            }
            else
            {
                while(!sk.empty() && height[i] > height[sk.top()])
                {
                    int tmp = sk.top();
                    sk.pop();
                    if(!sk.empty())
                    {
                        int h = min(height[i],height[sk.top()]) - height[tmp];
                        int w = i - sk.top() - 1;
                        ans += h*w;
                    }
                }
                sk.push(i);
            }
        }
        return ans;
    }
};