目录
739 每日温度
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
//栈里面放的是下标
stack<int> st;
//result放的是差值也就是结果
vector<int> result(temperatures.size(), 0);
st.push(0);
for(int i = 1; i < temperatures.size(); i++){
//1 栈元素比当前元素大,那我们就要把元素插进去
if(temperatures[st.top()] > temperatures[i]){
st.push(i);
}//2 栈元素等于当前元素,也要插进去
else if(temperatures[st.top()] == temperatures[i]){
st.push(i);
}//3 栈元素比当前元素小,说明我们找到要的元素了,就可以开始操作了
else{
//while是因为当前的元素可能比栈里面的好几个元素都大,那就说明这几个元素都找到了
while(!st.empty() && temperatures[st.top()] < temperatures[i]){
result[st.top()] = i - st.top();//赋值
st.pop();//把插进去的栈顶元素pop掉
}
//别忘记了把当前元素插进来
st.push(i);
}
}
return result;
}
};
496 下一个更大元素 I
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> unmp;
int index = 0;//记录nums2在nums1里面出现的位置
//把nums1里面的元素插进map做个记录,记一下对应元素出现的位置(下标),方便result进行插入
for(int i =0; i < nums1.size(); i++){
unmp[nums1[i]] = i;
}
//用单调栈的方法来找nums2里面元素的下一个更大元素
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()]){
//在map里面查询看看nums1里面是否存在当前nums2的值
if(unmp.count(nums2[st.top()]) > 0){
//如果存在,就算一下当前值在nums1的位置然后在result的对应位置插进去
index = unmp[nums2[st.top()]];
result[index] = nums2[i];
}
st.pop();
}
st.push(i);
}
}
return result;
}
};