其他分享
首页 > 其他分享> > 每日温度(leetcode)-堆栈

每日温度(leetcode)-堆栈

作者:互联网

每日温度(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;
    }
};

标签:下标,每日,元素,栈顶,st,vector,temperatures,堆栈,leetcode
来源: https://blog.csdn.net/elexiangchirou/article/details/121189255