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

 503.下一个更大元素II 

这道题和 739. 每日温度 几乎如出一辙，可以自己尝试做一做

代码随想录

这道题是循环数组，因此在遍历时要遍历nums.size()*2次，然后再读取nums中的元素时要使用i % nums.size()来得到对应的下标。

vector<int> nextGreaterElements(vector<int>& nums) {
        stack<int> st;
        vector<int> result(nums.size(), -1);
        if (nums.size() == 0) return result;

        for (int i = 0; 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());
        }
        return result;
    }

 42. 接雨水  

接雨水这道题目是 面试中特别高频的一道题，也是单调栈 应用的题目，大家好好做做。

建议是掌握 双指针 和单调栈，因为在面试中 写出单调栈可能 有点难度，但双指针思路更直接一些。

在时间紧张的情况有，能写出双指针法也是不错的，然后可以和面试官在慢慢讨论如何优化。 

代码随想录

这道题使用单调栈的方法，在遍历到的元素大于栈顶元素时，说明中间出现了低地，如果栈左边还有元素，说明该地左右都大于它，因此可以用来装雨水。在遍历过程中通过弹栈、比较栈中剩余元素的高度来得到装雨水的多少。

int trap(vector<int>& height) {
        int sum = 0;
        stack<int> st;
        st.push(0);
        for (int i = 0; i < height.size(); i++) {
            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;
                    sum += h * w;
                }
            }
            st.push(i);
        }
        return sum;
    }