这道题和?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;
}
接雨水这道题目是?面试中特别高频的一道题,也是单调栈?应用的题目,大家好好做做。
建议是掌握?双指针?和单调栈,因为在面试中?写出单调栈可能?有点难度,但双指针思路更直接一些。
在时间紧张的情况有,能写出双指针法也是不错的,然后可以和面试官在慢慢讨论如何优化。?
这道题使用单调栈的方法,在遍历到的元素大于栈顶元素时,说明中间出现了低地,如果栈左边还有元素,说明该地左右都大于它,因此可以用来装雨水。在遍历过程中通过弹栈、比较栈中剩余元素的高度来得到装雨水的多少。
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;
}
?