本文主要是介绍代码随想录训练营第五十九天| ● 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;}
这篇关于代码随想录训练营第五十九天| ● 503.下一个更大元素II ● 42. 接雨水的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!