代码随想录算法训练营第 48 天 |LeetCode 739. 每日温度 LeetCode496.下一个更大元素 I LeetCode503.下一个更大元素II

HIT最菜电控

于 2024-08-21 00:21:26 发布

阅读量191

点赞数 6

分类专栏：代码随想录算法训练营文章标签：算法 leetcode

本文链接：https://blog.csdn.net/2301_79647020/article/details/141337562

版权

代码随想录算法训练营专栏收录该内容

37 篇文章 0 订阅

订阅专栏

代码随想录算法训练营

Day48 代码随想录算法训练营第 48 天 |LeetCode 739. 每日温度 LeetCode496.下一个更大元素 I LeetCode503.下一个更大元素II

前言

一、基础

1、栈内存放下标

2、应用：找元素右边第一个大于/小于他的元素

3、找元素右边第一个大于他的元素

（1）a[0]:将0放入栈
（2）a[i]:
1）如果栈空：i入栈
2）如果栈顶对应元素大于等于a[i]:i入栈
3）栈顶对应元素小于a[i]：（while循环控制边界）
① a[i]就是栈顶对应元素右边第一个大于他的元素，记录答案
② 弹出栈顶
③ i入栈

二、LeetCode 739. 每日温度

1.题目链接

LeetCode 739. 每日温度

2.思路

1）a[0]:将0放入栈
（2）a[i]:
1）如果栈空：i入栈
2）如果栈顶对应元素大于等于a[i]:i入栈
3）栈顶对应元素小于a[i]：（while循环控制边界）
① a[i]就是栈顶对应元素右边第一个大于他的元素，记录答案---- i-s.top()
② 弹出栈顶
③ i入栈

3.题解

class Solution {
public:
    stack<int> s;
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> res(n, 0);
        s.push(0);
        for (int i = 1; i < n; i++) {
            if (s.size() == 0)
                s.push(i);
            else {
                if (temperatures[s.top()] >= temperatures[i]) {
                    s.push(i);
                } else {
                    while (!s.empty() &&
                           temperatures[s.top()] < temperatures[i]) {
                        res[s.top()] = i - s.top();
                        s.pop();
                    }
                    s.push(i);
                }
            }
        }
        return res;
    }
};

三、LeetCode496.下一个更大元素

1.题目链接

LeetCode496.下一个更大元素

2.思路

问题转换：
问题转换成：同时属于nums1和nums2的元素在nums2中右边第一个比自己大的元素下标
方法一：
（1）单调栈存放待处理的元素下标：
单调栈存放nums2的元素在Nums2中下标（但是真正对答案有用的是同时属于nums1和nums2的元素）
（2）过程
1）为了通过同时属于nums1和nums2的元素值找在nums1中的下标，用映射，元素为key，下标为values
2）单调栈
① 下标0放入
② 栈空或栈顶元素>=nums2[i]：入栈
③ 栈顶元素<nums2[i]:
如果栈顶元素在nums1出现，那么栈顶元素在nums1中对应元素的答案是nums2[i]
无论栈顶元素是否在nums1出现，弹出栈顶
跳出循环后i入栈
方法二：
1）单调栈存放待处理的元素下标：
单调栈存放同时属于nums1和nums2的元素在Nums2中下标
（2）过程
1）为了通过同时属于nums1和nums2的元素值找在nums1中的下标，用映射，元素为key，下标为values
2）单调栈
① 栈空或栈顶元素>=nums2[i]：如果nums2[i]在nums1出现，i入栈
③ 栈顶元素<nums2[i]:
栈顶元素在nums1中对应元素的答案是nums2[i]
弹出栈顶
跳出循环后，如果nums2[i]在nums1出现，i入栈

3.题解

方法1

class Solution {
public:
stack<int>s;
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n1=nums1.size();
        int n2=nums2.size();
        vector<int>res(n1,-1);
        s.push(0);
        unordered_map<int,int>map;
        for(int i=0;i<n1;i++){
            map[nums1[i]]=i;
        }
        for(int i=1;i<n2;i++)
        {
            if(s.size()==0 || nums2[s.top()]>=nums2[i])s.push(i);
            else
            {
                while(!s.empty()&&nums2[s.top()]<nums2[i])
                {
                    if(map.count(nums2[s.top()]) > 0)
                        res[map[nums2[s.top()]]]=nums2[i];
                    s.pop();
                }
                
                s.push(i);
            }

        }
        return res;
    }
};

方法2

class Solution {
public:
    stack<int> s;
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n1 = nums1.size();
        int n2 = nums2.size();
        vector<int> res(n1, -1);

        unordered_map<int, int> map;
        for (int i = 0; i < n1; i++) {
            map[nums1[i]] = i;
        }
        for (int i = 0; i < n2; i++) {
            if ((s.size() == 0 || nums2[s.top()] >= nums2[i]) &&
                map.count(nums2[i]) > 0) {
                s.push(i);

            } else {
                while (!s.empty() && nums2[s.top()] < nums2[i]) {

                    res[map[nums2[s.top()]]] = nums2[i];
                    s.pop();
                }

                if (map.count(nums2[i]) > 0)
                    s.push(i);
            }
        }
        return res;
    }
};

四、LeetCode503.下一个更大元素II

1.题目链接

LeetCode503.下一个更大元素II

2.思路

（1）审题：循环数组

e.g.输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2

（2）为了确保每个元素都能搜索完一个周期：从0到2n-1遍历
（相当于将数组扩增：abd—abdabd）

3.题解

class Solution {
public:
    stack<int> s;
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n = nums.size();
        vector<int> res(n, -1);

        for (int i = 0; i < n * 2; i++) {
            if (s.empty() || nums[s.top()] >= nums[i % n])
                s.push(i % n);
            else {
                while (!s.empty() && nums[s.top()] < nums[i % n]) {
                    res[s.top()] = nums[i % n];
                    s.pop();
                }
                s.push(i % n);
            }
        }
        return res;
    }
};