训练营打卡Day62
思路
- 用一个递增栈存储每天的温度下标,遍历每一天的温度。
- 如果当前天的温度大于栈顶天的温度,说明当前天比栈顶天温度高,那么更新栈顶天到当前天的天数,弹出栈顶天,循环此过程直到栈为空或者当前天的温度小于等于栈顶天的温度。
- 最后将当前天的下标入栈。最终的结果是一个每天到比它温度高的第一天的天数的数组。
代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
stack<int> st;
vector<int> result(T.size(), 0);
for (int i = 0; i < T.size(); i++) {
while (!st.empty() && T[i] > T[st.top()]) {
result[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
return result;
}
};
思路
- 使用一个栈来辅助查找,并使用一个unordered_map将元素映射到nums1中的位置。
- 对于nums2中的每个元素,如果栈顶元素小于该元素,则弹出栈顶元素并在unordered_map中查询该元素是否在nums1中,如果存在,则将该元素的索引位置的结果更新为该元素的值。最后,将当前元素压入栈中。
- 最后,返回结果数组,其中包含每个元素在nums2中的下一个更大的元素的值,如果没有更大的元素,则为-1。
代码
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
stack<int> st;
vector<int> result(nums1.size(), -1);
if (nums1.size() == 0) return result;
unordered_map<int, int> umap;
for (int i = 0; i < nums1.size(); i++) {
umap[nums1[i]] = i;
}
st.push(0);
for (int i = 1; i < nums2.size(); i++) {
while (!st.empty() && nums2[i] > nums2[st.top()]) {
if (umap.count(nums2[st.top()]) > 0) {
int index = umap[nums2[st.top()]];
result[index] = nums2[i];
}
st.pop();
}
st.push(i);
}
return result;
}
};