对于单调栈的应用情况是求右边(左边)第一个大于或者小于该元素的值或者角标。右边大于则栈顶到栈底单调递增,右边小于则栈顶到栈底单调递减。具体方法是第一个元素直接入栈,遍历从1的元素开始,如果当前值小于等于栈顶元素,直接入栈;否则进行一个循环遍历,对于栈顶每一个小于当前元素,计算距离,并且出栈,直到全部满足要求出栈,再把当前值入栈。返回结果。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
// vector<int> res(temperatures.size(), 0);
// for (int i = 0; i < temperatures.size(); i++) {
// for (int j = i+1; j < temperatures.size(); j++) {
// if (temperatures[j] > temperatures[i]) {
// res[i] = j-i;
// break;
// }
// }
// }
// return res;
stack<int> st;
vector<int> res(temperatures.size(), 0);
// 第一个元素无脑入栈
st.push(0);
for (int i = 1; i < temperatures.size(); i++) {
// res 存储的是角标
if (temperatures[i] > temperatures[st.top()]) {
// 如果当前值大于栈顶元素,一直出栈直到栈空或者不大于栈顶
while(!st.empty() && temperatures[i] > temperatures[st.top()]){
res[st.top()] = i-st.top();
st.pop();
}
// 入栈
st.push(i);
}
// 小于等于栈顶元素直接入栈
else {
st.push(i);
}
}
return res;
}
};
这题题目太绕了,但是意思就是一组元素在另一组元素里,要保留前者的顺序,还要找到前者在后者中的最近较大元素。采用哈希和单调栈实现,其实暴力实现应该也能通过,但是能用好方法当然就用,就算记忆也是好的。mad算法题刷不动了,一边刷一边不会,老子要开始背了。
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
// 哈希map加单调栈,单调栈就是右边第一个大于左边或者右边第一个小于左边
stack<int> st;
vector<int> res(nums1.size(), -1);
st.push(0);
unordered_map<int ,int> umap;
for (int i = 0; i < nums1.size(); i++) {
umap[nums1[i]] = i;
}
for (int i = 0; i < nums2.size(); i++) {
if (nums2[i] <= nums2[st.top()]) {
st.push(i);
}
else {
while (!st.empty() && nums2[i] > nums2[st.top()]) {
if (umap.find(nums2[st.top()]) != umap.end()) {
int index = umap[nums2[st.top()]];
res[index] = nums2[i];
}
st.pop();
}
st.push(i);
}
}
return res;
}
};