每日温度
如果我们单纯的遍历数组,我们不知道当前元素是否比之前的元素大,所以需要维护一个容器来记录遍历过的元素。
什么时候用单调栈?通常是一维数组,要寻找任一个元素的右边或左边第一个比自己大或小的元素的位置。时间复杂度为O(n)。
这道题的单调栈应该是递增栈还是递减栈呢(从栈头到栈底)?这道题是求右边第一个比自己大的元素的位置,小于等于栈顶元素时当前元素下标应该入栈;遇到大于栈顶元素的情况,应该记录当前元素与栈顶元素下标差,result[st.top()] = i - st.top()
,并弹出栈顶元素,重复操作直到当前元素小于等于栈顶元素,此时将当前元素下标入栈。
- 情况一:当前遍历的元素T[i]大于栈顶元素T[st.top()]
- 情况二:当前遍历的元素T[i]等于栈顶元素T[st.top()]
- 情况三:当前遍历的元素T[i]小于栈顶元素T[st.top()]
由此可知,求右边第一个比自己大的元素的位置,应该维护从栈头到栈底的递增栈。
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {vector<int> result(temperatures.size(), 0);stack<int> st;st.push(0);for(int i = 1; i < temperatures.size(); i++) {while(!st.empty() && temperatures[i] > temperatures[st.top()]) {result[st.top()] = i - st.top();st.pop();}// 此时当前温度一定是小于等于栈顶元素,当前元素下标入栈st.push(i);}return result;}
};
下一个更大元素I
相对于上一道题,这道题搞了两个数组,数组1是数组2的子集,这一点有点绕,因为最终求的结果与数组1长度一致,所以我们应该构建数组1长度的结果数组,并根据题意初始化为-1。
另外要注意是找数组1中的元素在数组2中的下一个更大元素,所以单调栈的操作应该是在遍历数组2的过程中进行的,入栈出栈操作与上一题相同,只是在记录结果数组时需要判断栈顶元素是否在数组1中。
判断一个元素是否在数组中,很容易想到哈希表,建立数值到下标的映射,更方便地记录结果。
class Solution {
public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {vector<int> result(nums1.size(), -1);stack<int> st;unordered_map<int, int> umap;for(int i = 0; i < nums1.size(); i++) { // 建立数组1中数值到下标的映射umap[nums1[i]] = i;}for(int i = 0; i < nums2.size(); i++) { // 遍历数组2while(!st.empty() && nums2[i] > nums2[st.top()]) { // 当前元素大于栈顶元素if(umap.count(nums2[st.top()]) > 0) { // 如果栈顶元素出现在数组1中,结果应该被记录int index = umap[nums2[st.top()]];result[index] = nums2[i];}st.pop(); // 不论是否结果是否记录,必须维护单调栈}// 当前元素一定小于等于栈顶元素,满足递增栈条件(从栈头到栈底)st.push(i);}return result;}
};