代码随想录算法训练营第五十八天|739. 每日温度 496.下一个更大元素 I

目录

LeeCode 739. 每日温度 

LeeCode 496.下一个更大元素 I  

---

LeeCode 739. 每日温度 

739. 每日温度 - 力扣（LeetCode）

【单调栈】

一维数组寻找任一元素左边或右边第一个比它大或小的元素的位置时，可以使用单调栈，用一个栈记录遍历过的元素，栈中存放元素下标。求一个元素右边第一个更小的元素时，栈顶到栈底递减，求更大元素时，栈顶到栈底递增排列。

【过程图示】

 【代码实现】

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& t) {
    	stack<int> st;
		vector<int> result(t.size(), 0);
		st.push(0);
		for (int i = 1; i < t.size(); i++) {
			if (t[i] < t[st.top()]) st.push(i);
			else if (t[i] == t[st.top()]) st.push(i);
			else {
				while (!st.empty() && t[i] > t[st.top()]) {
					result[st.top()] = i - st.top();
					st.pop();
				}
				st.push(i); 
			}
		} 
        return result;
    }
};

时间复杂度：O（n）                                                            空间复杂度：O（n）

---

LeeCode 496.下一个更大元素 I  

496. 下一个更大元素 I - 力扣（LeetCode）

【思路】

定义一个和nums1大小的数组result 来存放结果，初始化为-1；两个数组无重复元素，故可以使用map 做映射；使用单调栈，栈顶到栈底递增。

【代码】

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++) {
			if (nums2[i] <= nums2[st.top()])  st.push(i);
			else {
				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;
    }
};