文章目錄
- 1. 739. 每日溫度
- 2. 496.下一個更大元素 I
1. 739. 每日溫度
參考文檔:代碼隨想錄
分析:
找下一個更高溫度出現在幾天后,即當前位置右側出現的一個比它更大的值,如果是暴力搜索,兩層for,時間復雜度O(n2), 數據范圍105,時間最大是1010,這可能會超,我試了試,確實超了,用一個很大的全都一樣的數組就超了。
單調棧是棧里是數組的下標,下標對應的數組元素是有序的。這里使用從棧頂到棧底遞增的方法:這樣遇到比棧頂更大的就可以根據下標求出距離了。遇到的元素和棧頂的元素會有三種情況:
棧頂元素大于遍歷元素:入棧保留。
棧頂元素等于遍歷元素:入棧保留。
棧頂元素小于遍歷元素:彈出棧頂,下標相減再減一求出距離。根據棧里的單調,當前遍歷元素和棧頂一直抵消,直到棧頂元素大于遍歷元素。
代碼:
class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {vector<int> answer(temperatures.size(), 0);stack<int> st;//遞增棧,存放的是下標st.push(0);for(int i = 1; i < temperatures.size(); i++){if(temperatures[i] < temperatures[st.top()]){//小于棧頂st.push(i);}else if(temperatures[i] == temperatures[st.top()]){st.push(i);}else{while(!st.empty() && temperatures[i] > temperatures[st.top()]){answer[st.top()] = i - st.top();st.pop();}st.push(i);} }if(!st.empty()){answer[st.top()] = 0;st.pop();}return answer;}
};
2. 496.下一個更大元素 I
參考文檔:代碼隨想錄
分析:
和溫度一樣要找出右側第一個比當前元素大的值,但是在找的時候需要查詢當前元素在nums1中在不在,在的話保存值,而不是距離。因為用到查詢是不是在nums1中存在,unordered_map<int, int>哈希表的底層邏輯查找效率是O(1),保存nums1中的數組下標和數組元素。
代碼:
class Solution {
public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {stack<int> st;vector<int> answer(nums1.size(), -1);if(nums1.size() == 0) return answer;unordered_map<int, int> umap;for(int i = 0; i < nums1.size(); i++){umap[nums1[i]] = i;//保存值和下標,下標用來更新result}st.push(0);for(int i = 1; i < nums2.size(); i++){if(nums2[i] < nums2[st.top()]){st.push(i);}else 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) answer[umap[nums2[st.top()]]] = nums2[i];st.pop();}st.push(i);}}return answer;}
};