1 单调栈 leetcode496 下一个更大元素

题目描述

给定两个没有重复元素的数组 nums1nums2 ,其中nums1nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。nums1 中数字 x的下一个更大元素是指 xnums2 中对应位置的右边的第一个比 x大的元素。如果不存在,对应位置输出-1。

示例 1:

输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
    对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。
    对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。
    对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。

示例2:

输入: nums1 = [2,4], nums2 = [1,2,3,4].
输出: [3,-1]
解释:
    对于num1中的数字2,第二个数组中的下一个较大数字是3。
    对于num1中的数字4,第二个数组中没有下一个更大的数字,因此输出 -1。

注意:

  1. nums1nums2中所有元素是唯一的。
  2. nums1nums2 的数组大小都不超过1000。

官方难度

Easy

解决思路

最简单粗暴的解决思路,两层循环:
输入: nums1 = [2,4], nums2 = [1,2,3,4].

  1. 遍历nums1,得到当前元素i2.
  2. 遍历nums2,得到元素 i (2) 在当前nums2中的下标start.
  3. start开始找到nums2大于i (2)的元素 3

优化

这个题目考的是单调栈, 栈中存的是递减的数字,即当前输入元素是否大于栈顶元素,如果大于则出栈,否则将当前元素入栈。因为只有在当前元素大于栈顶元素的时候才出栈,因此,当出栈的时候,当前元素即为栈顶元素的下一个最更大值,可以构造出栈元素与下一个最大元素的映射,。以输入[1,9,3,2,5]为例子.

  1. 初始化记忆字典memo, 并且1入栈
  2. 下一个元素9判断是否大于栈顶元素1,大于则出栈,memo[1] = 9 并且将9入栈;
  3. 下一个3判断是否大于栈顶元素9, 小于则入栈。
  4. 下一个2判断是否大于栈顶元素3, 小于则入栈。
  5. 下一个5判断是否大于栈顶元素2, 大于则出栈, memo[2] = 5
  6. 继续判断下一个5判断是否大于栈顶元素3, 大于则出栈, memo[3] = 5
  7. 继续判断下一个5判断是否大于栈顶元素9, 小于则入栈。

最终栈的结果为 [9, 5],遍历nums1,如果当前元素不在memo中则添加-1,否则添加memo中存储的结果。

根据上面的分析,我们可以很容易的得到直接方案的流程如下:

  1. 遍历nums2, 通过单调递减栈构造记忆函数memo
  2. 遍历nums1,如果当前元素不在memo中则添加-1,否则添加memo中存储的结果

基于这个流程,我们可以实现类似下面的代码:

class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        res = []
        stack = []
        memo = {}
        for i in nums2:
            while stack and i > stack[-1]:
                memo[stack[-1]] = i
                stack.pop()
            stack.append(i)
        for j in nums1:
            if j in memo:
                res.append(memo[j])
            else:
                res.append(-1)
        return res

总结

这题为Easy难度,大部分人都能想到暴力破解,但是如果能在面试过程中,不断优化,加入单调栈的思想,面试过程中是可以加分的。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,098评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,213评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,960评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,519评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,512评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,533评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,914评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,574评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,804评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,563评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,644评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,350评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,933评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,908评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,146评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,847评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,361评论 2 342

推荐阅读更多精彩内容