代码随想录训练营第五十九天｜503.下一个更大元素II，42. 接雨水

503.下一个更大元素II

题目链接：力扣

代码：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        stack<int> st;
        vector<int> result(nums.size()*2,-1);
        st.push(0);
        for(int i = 1; i < nums.size(); i++)
        {
            while(!st.empty() && nums[i] > nums[st.top()])
            {
                result[st.top()] = nums[i];
                st.pop();
            }
            st.push(i);
        }
        for(int i = nums.size(); i < nums.size() * 2; i++)
        {
            while(!st.empty() && nums[i-nums.size()] > nums[st.top()])
            {
                result[st.top()] = nums[i-nums.size()];
                st.pop();
            }
            st.push(i-nums.size());
        }
        vector<int> reback(nums.size(),0);
        for(int i = 0; i < nums.size(); i++)
        {
            reback[i] = result[i];
        }
        return reback;
    }
};

代码：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        stack<int> st;
        vector<int> result(nums.size(),-1);
        st.push(0);
        for(int i = 1; i < 2*nums.size(); i++)
        {
            while(!st.empty() && nums[i%nums.size()] > nums[st.top()])
            {
                result[st.top()] = nums[i%nums.size()];
                st.pop();
            }
            st.push(i%nums.size());
        }
        return result;
    }
};

更精简了

42. 接雨水

题目链接：力扣

代码（双指针法）：

class Solution {
public:
    int trap(vector<int>& height) {
        vector<int> leftHight(height.size(),0);
        vector<int> rightHight(height.size(),0);
        int size = height.size();
        leftHight[0] = height[0];
        for(int i = 1; i < height.size(); i++)
        {
            leftHight[i] = max(leftHight[i-1],height[i]);
        }
        rightHight[size - 1] = height[size-1];
        for(int i = size-2; i >= 0; i--)
        {
            rightHight[i] = max(rightHight[i+1],height[i]);
        }
        int result = 0;
        for(int i = 0; i < size; i++)
        {
            int count = min(leftHight[i],rightHight[i]) - height[i];
            if(count > 0)
                result += count;
        }
        return result;
    }
};

这个是按行来的，左边的最大和右边的最大取最小然后减去当前高度 如果不小于 说明是凹进去的 给累加一下

代码（单调栈）：

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> st;
        st.push(0);
        int result = 0;
        for(int i = 1; i < height.size(); i++)
        {
            if(height[i] < height[st.top()])
                st.push(i);
            else if(height[i] == height[st.top()])
            {
                st.pop();
                st.push(i);
            }
            else if(height[i] > height[st.top()])
            {
                while(!st.empty() && height[i] > height[st.top()])
                {
                    int mid = st.top();
                    st.pop();
                    if(!st.empty())
                    {
                        //比较左右两边的高度取小的那个 然后和当前高度求差 算出高度差
                        int h = min(height[i],height[st.top()]) - height[mid];  
                        int w = i - st.top() - 1;
                        result += h*w;
                    }
                }
                st.push(i);
            }
        }
        return result;
    }
};

上面那个是按列求和这个是按行求和 一旦栈顶比当前小，则说明有凹槽，然后找栈顶左右两边的min和栈顶求差，就是栈顶这一部分的高，然后求个宽