每日温度(leetcode)-堆栈
问题描述:
请根据每日 气温 列表 temperatures ,请计算在每一天需要等几天才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。
解法1-暴力求解
不用堆栈知识,提交时间会超限
class Solution {
public:
vector<int> dailyTemperatures(vector<int> &temperatures) {
vector<int> v;
int j;
//设两个指针,一个指针指向固定值,另一个元素指向要比较元素
//一找到比固定元素大就结束,比较下一个
for (int i = 0; i < temperatures.size(); i++) {
for (j = i + 1; j < temperatures.size(); j++) {
if (temperatures[j] > temperatures[i]) {
v.push_back(j - i);
break;
}
}
//没有找到说明没有比他大的,是0
if (j == temperatures.size()) {
v.push_back(0);
}
}
return v;
}
};
解法2-用堆栈存储
思路:用栈存储元素下标,先将第一个元素下标入栈,此后每个元素都与栈顶元素的值(是所代表的值,不是元素下标)比较。比栈顶元素所代表的值大,将将栈顶元素出栈,栈顶元素的值为栈顶元素与当前元素差值,一直循环到栈为空或者是栈顶元素比本元素大,将本元素入栈
class Solution {
public:
vector<int> dailyTemperatures( vector<int> &temperatures ) {
vector<int> v; //vector 返回值
stack<int> st; //栈 存元素下标
int arr[100005] = {0};
st.push( 0 ); //先将第一个元素下标入栈
for( int i = 1; i < temperatures.size(); i++ ) {
//当栈不为空且当前元素比栈顶元素大时 注意:st.top()也是元素的下标值
while( !st.empty() && temperatures[i] > temperatures[st.top()] ) {
//因为要随时存储不同元素值,我们先用数组存储
//栈顶元素的那个下标值所在数组值=两个下标差值 就是差多少天
arr[st.top()] = i - st.top();
st.pop(); //赋值完成,栈顶元素出栈
}
//栈为空 或者 比栈顶元素小 入栈
st.push( i );
}
//我们之前将数组初始化0是因为后面元素可能没有比他大的,像最后一个元素,肯定是0
for( int i = 0; i < temperatures.size(); i++ ) {
v.push_back( arr[i] );
}
return v;
}
};