代码随想录算法训练营29期Day61|LeetCode 739,496

文档讲解:每日温度 下一个更大元素I

739.每日温度

题目链接https://leetcode.cn/problems/daily-temperatures/description/

思路:

维护一个单调递减的栈就行了。

一次读取一个数组中的元素,将其与栈顶元素比较,如果比栈顶元素大,证明找到了栈顶元素右侧第一个比它大的,记录并弹出栈顶即可。

重复上述比较直至该元素小于栈顶元素或者栈空。

加入这个元素即可。

重复上述操作可解决问题。

核心代码:

cpp 复制代码
class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int> st;
        int n=temperatures.size();
        vector<int> ans(n,0);
        for(int i=0;i<n;i++){
            while(!st.empty()&&temperatures[st.top()]<temperatures[i]){
                ans[st.top()]=i-st.top();
                st.pop();
            }
            st.push(i);
        }
        return ans;
    }
};

496.下一个更大元素I

题目链接: https://leetcode.cn/problems/next-greater-element-i/description/

思路:

这题有O()的做法,就是枚举nums1中的数字,去nums2中遍历找到其位置,然后再向后找第一个比它大的值即可。

这种方法很简单,也能过这道题数据范围,但不在此赘述。

下面阐述一种O(n)的做法:

对nums2使用单调栈,维护一个单调递减的序列(详情见上一道题目),由此可得到nums2中每个值右侧的第一个比它大的值,记为map。

然后遍历nums1,根据map确定nums1中的值的下一个更大元素,获得答案数组输出即可。

核心代码:

cpp 复制代码
class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> mp;
        stack<int> st;
        int n=nums2.size();
        vector<int> ans(nums1.size(),-1);
        for(int i=0;i<n;i++){
            while(!st.empty()&&nums2[st.top()]<nums2[i]){
                mp[nums2[st.top()]]=nums2[i];
                st.pop();
            }
            st.push(i);
        }
        while(!st.empty()){
            mp[nums2[st.top()]]=-1;
            st.pop();
        }
        n=nums1.size();
        for(int i=0;i<n;i++) ans[i]=mp[nums1[i]];
        return ans;
    }
};

今日总结

这次的题学习时长1h,挺简单的。

接着论文idea,头大。

相关推荐
Moonbit7 分钟前
月报Vol.03: 新增Bitstring pattern支持,构造器模式匹配增强
后端·算法·github
快手技术11 分钟前
多模态大模型Keye-VL-1.5发布!视频理解能力更强!
算法
JosieBook21 分钟前
【程序人生】有梦想就能了不起,就怕你没梦想
程序人生·职场和发展
薛定谔的算法26 分钟前
JavaScript数组操作完全指南:从基础到高级
前端·javascript·算法
可爱的小小小狼29 分钟前
算法:位运算
算法
zzzsde1 小时前
【数据结构】强化训练:从基础到入门到进阶(1)
数据结构
VisionPowerful1 小时前
九.弗洛伊德(Floyd)算法
算法·c#
可爱的小小小狼1 小时前
算法:哈希表
redis·算法·散列表
奔跑吧 android1 小时前
【linux kernel 常用数据结构和设计模式】【数据结构 3】【模拟input子系统input_dev和input_handler之间的多对多关系】
linux·数据结构·input·kernel·input_dev·input_handler·input_handle
点云侠1 小时前
解决Visual Studio 2022编译工程速度慢的问题
开发语言·c++·ide·算法·计算机视觉·visual studio