1.739. 每日温度 - 力扣(LeetCode)
用单调栈的方法做:
从左到右遍历数组:
栈中存放的是下标,每个温度在原数组中的下标,从大到小排列,因为这样才能确保的是最近一天的升高温度
如果栈为空,则直接将i进栈;当此时的栈不为空,并且此时遍历的温度大于栈顶的温度(即栈中最大的温度)循环以下操作:将当前的温度下标减去栈顶元素的下标,然后栈顶元素的下标就是此时所计算的那一天
res[st.top()] = i - st.top();
这意味着从索引st.top()代表的那一天开始,直到现在的索引i,我们首次遇到了一个更高的温度,因此,我们可以确定在st.top()这一天之后的i - st.top()天,有一个更高的温度出现。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> res(n, 0);
stack<int> st;
for(int i = 0; i < n; i++){
while(!st.empty() && temperatures[i] < temperatures[st.top()]{
res[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
return res;
}
};
2.496. 下一个更大元素 I - 力扣(LeetCode)
这中下一个更大更小元素的就是用单调栈
首先注意是要根据nums1的元素,去nums2中查找对应的下一个更大元素,所以,返回数组的大小应该是跟nums1保持一致的
从右至左到序遍历数组:
如果栈不为空并且,此时遍历的数字大于栈顶中的数字,我们就可以把栈中比此时数字小的数字弹出,因为我们要找的是下一个比它大的数字,比他小的肯定不行啊。
循环结束的两种条件,栈为空,找到比它大的数字了
如果是前者的话,代表这个数组里面没有比这个数组更大的数字了,就-1
如果是后者的话,我们就找到一个当前数组里比他大的下一个元素,最近的。就把这个比它大的数字压入栈中呗。
nums1中的元素必定在nums2中,我们在哈希表中保存nums2中每个数字的下一个更大的元素,最后根据nums1元素访问就好
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
unordered_map<int, int> map;
stack<int> st;
vector<int> res(nums1.size());
for(int i = nums2.size() - 1; i >= 0; i--){
int num = nums2[i];
while(!st.empty() && num >= st.top()){
st.pop();
}
map[num] = st.empty() ? -1 : st.top();
st.push(num);
}
for(int i = 0; i < nums1.size(); i++){
res[i] = map[nums1[i]];
}
return res;
}
};
首先用暴力的做法还是弯弯绕绕费劲做出来了
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size(), m = nums2.size();
vector<int> res;
int k = 0;
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
if(nums1[i] == nums2[j]){
for(k = j + 1; k < m; k++){
if(nums2[j] < nums2[k]){
res.push_back(nums2[k]);
break;
}
}
if(k >= m){
res.push_back(-1);
}
}
}
}
return res;
}
};