力扣每日一题 6/24 模拟数组单调栈

誓则盟约

于 2024-06-24 11:01:54 发布

阅读量1.2k

点赞数 37

分类专栏： IT竞赛文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/Xxy_1008/article/details/139918827

版权

IT竞赛专栏收录该内容

47 篇文章 6 订阅

订阅专栏

博客主页：誓则盟约
系列专栏：IT竞赛专栏
关注博主，后期持续更新系列文章
如果有错误感谢请大家批评指出，及时修改
感谢大家点赞👍收藏⭐评论✍

503.下一个更大元素II 【中等】

题目：

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。

示例 1:

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

示例 2:

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

提示:

1 <= nums.length <= 10**4
-10**9 <= nums[i] <= 10**9

分析问题：

思路1：

首先，通过两次遍历数组（实际是通过对索引取模来模拟）。在每次遍历中，如果当前元素大于栈顶元素所对应的数组值，就将栈顶元素弹出，并在结果数组中记录当前元素为栈顶元素的下一个更大元素。

在前 n 个位置时，将索引存入栈中。如果在后续的遍历中找到了更大的元素，就进行相应的处理。

最终，结果数组 ans 中存储了每个元素对应的下一个更大元素，如果没有则为 -1 。

思路2：

模拟。根据题目给的要求，我们遍历整个数组，去寻找他下一个更大的值，这里要把数组变成循环数组，因为我们要遍历整个列表。变成循环数组的方法无非就是把遍历的长度加长一点，然后模上len（nums），时间复杂度是O(N**2),但是这个方法用时间换空间，空间复杂度很低。

代码实现：

思路1：

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        n = len(nums)
        stack = []
        ans = [-1]*n 

        for i in range(2*n):
            x = nums[i % n]
            while stack and x > nums[stack[-1]]:
                ans[stack.pop()] = x 
            if i < n:
                stack.append(i)
        return ans

这种方法时间复杂度和空间复杂度都是比较优越的。

思路2：

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        n=len(nums)
        ls=[]
        for i in range(len(nums)):
            for j in range(i+1,i+len(nums)+1):
                if j%n==i: 
                    ls.append(-1)
                    break
                elif nums[j%n] > nums[i]:
                    ls.append(nums[j%n])
                    break
        return ls

思路2是比较容易想出来的一种方法，他的优点就在于空间复杂度较低，通俗易懂。但是时间复杂度较高。

总结：

思路一（使用栈）代码详解：

首先，获取输入数组 nums 的长度 n ，并初始化一个空栈 stack 和结果数组 ans ，其中 ans 中的每个元素初始值都为 -1 。
然后，通过一个循环，模拟对数组进行两轮遍历（实际是通过 i % n 来实现取模操作，达到循环数组的效果）。
对于每次循环获取的当前元素 x ，如果栈不为空且 x 大于栈顶元素对应的数组值，就将栈顶元素弹出，并在结果数组 ans 中记录 x 为弹出元素的下一个更大元素。
在前 n 次循环中（即第一次遍历数组时），将当前索引 i 存入栈中。
最终，返回结果数组 ans ，其中每个位置存储了对应元素的下一个更大元素，如果没有则为 -1 。

思路二（使用两层循环）代码详解：

首先获取输入数组 nums 的长度 n ，并创建一个空列表 ls 用于存储结果。
外层循环遍历数组中的每个元素。对于每个元素，通过内层循环从其右侧的下一个位置开始，模拟循环数组的方式进行查找。
在内层循环中，如果当前位置经过取模后又回到了起始位置（即 j % n == i ），说明在右侧没有找到更大的元素，将 -1 存入结果列表 ls 并结束内层循环。
如果找到了右侧第一个比当前元素大的元素（即 nums[j % n] > nums[i] ），将该更大元素存入结果列表 ls 并结束内层循环。
外层循环结束后，返回结果列表 ls 。