|
| 1 | + |
| 2 | +# 503.下一个更大元素II |
| 3 | + |
| 4 | +链接：https://leetcode-cn.com/problems/next-greater-element-ii/ |
| 5 | + |
| 6 | +给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。 |
| 7 | + |
| 8 | +示例 1: |
| 9 | + |
| 10 | +* 输入: [1,2,1] |
| 11 | +* 输出: [2,-1,2] |
| 12 | +* 解释: 第一个 1 的下一个更大的数是 2；数字 2 找不到下一个更大的数；第二个 1 的下一个最大的数需要循环搜索，结果也是 2。 |
| 13 | + |
| 14 | + |
| 15 | +# 思路 |
| 16 | + |
| 17 | +做本题之前建议先做[739. 每日温度](https://mp.weixin.qq.com/s/YeQ7eE0-hZpxJfJJziq25Q) 和 [496.下一个更大元素 I](https://mp.weixin.qq.com/s/U0O6XkFOe-RMXthPS16sWQ)。 |
| 18 | + |
| 19 | +这道题和[739. 每日温度](https://mp.weixin.qq.com/s/YeQ7eE0-hZpxJfJJziq25Q)也几乎如出一辙。 |
| 20 | + |
| 21 | +不同的时候本题要循环数组了。 |
| 22 | + |
| 23 | +关于单调栈的讲解我在题解[739. 每日温度](https://mp.weixin.qq.com/s/YeQ7eE0-hZpxJfJJziq25Q)中已经详细讲解了。 |
| 24 | + |
| 25 | +本篇我侧重与说一说，如何处理循环数组。 |
| 26 | + |
| 27 | +相信不少同学看到这道题，就想那我直接把两个数组拼接在一起，然后使用单调栈求下一个最大值不就行了！ |
| 28 | + |
| 29 | +确实可以！ |
| 30 | + |
| 31 | +讲两个nums数组拼接在一起，使用单调栈计算出每一个元素的下一个最大值，最后再把结果集即result数组resize到原数组大小就可以了。 |
| 32 | + |
| 33 | +代码如下： |
| 34 | + |
| 35 | +```C++ |
| 36 | +// 版本一 |
| 37 | +class Solution { |
| 38 | +public: |
| 39 | + vector<int> nextGreaterElements(vector<int>& nums) { |
| 40 | + // 拼接一个新的nums |
| 41 | + vector<int> nums1(nums.begin(), nums.end()); |
| 42 | + nums.insert(nums.end(), nums1.begin(), nums1.end()); |
| 43 | + // 用新的nums大小来初始化result |
| 44 | + vector<int> result(nums.size(), -1); |
| 45 | + if (nums.size() == 0) return result; |
| 46 | + |
| 47 | + // 开始单调栈 |
| 48 | + stack<int> st; |
| 49 | + for (int i = 0; i < nums.size(); i++) { |
| 50 | + while (!st.empty() && nums[i] > nums[st.top()]) { |
| 51 | + result[st.top()] = nums[i]; |
| 52 | + st.pop(); |
| 53 | + } |
| 54 | + st.push(i); |
| 55 | + } |
| 56 | + // 最后再把结果集即result数组resize到原数组大小 |
| 57 | + result.resize(nums.size() / 2); |
| 58 | + return result; |
| 59 | + } |
| 60 | +}; |
| 61 | + |
| 62 | +``` |
| 63 | + |
| 64 | +这种写法确实比较直观，但做了很多无用操作，例如修改了nums数组，而且最后还要把result数组resize回去。 |
| 65 | + |
| 66 | +resize倒是不费时间，是O(1)的操作，但扩充nums数组相当于多了一个O(n)的操作。 |
| 67 | + |
| 68 | +其实也可以不扩充nums，而是在遍历的过程中模拟走了两边nums。 |
| 69 | + |
| 70 | +代码如下： |
| 71 | + |
| 72 | +```C++ |
| 73 | +// 版本二 |
| 74 | +class Solution { |
| 75 | +public: |
| 76 | + vector<int> nextGreaterElements(vector<int>& nums) { |
| 77 | + vector<int> result(nums.size(), -1); |
| 78 | + if (nums.size() == 0) return result; |
| 79 | + stack<int> st; |
| 80 | + for (int i = 0; i < nums.size() * 2; i++) { |
| 81 | + // 模拟遍历两边nums，注意一下都是用i % nums.size()来操作 |
| 82 | + while (!st.empty() && nums[i % nums.size()] > nums[st.top()]) { |
| 83 | + result[st.top()] = nums[i % nums.size()]; |
| 84 | + st.pop(); |
| 85 | + } |
| 86 | + st.push(i % nums.size()); |
| 87 | + } |
| 88 | + return result; |
| 89 | + } |
| 90 | +}; |
| 91 | +``` |
| 92 | +
|
| 93 | +可以版本二不仅代码精简了，也比版本一少做了无用功！ |
| 94 | +
|
| 95 | +## 其他语言版本 |
| 96 | +
|
| 97 | +Java: |
| 98 | +
|
| 99 | +Python: |
| 100 | +
|
| 101 | +Go: |
| 102 | +
|
| 103 | +JavaScript: |
0 commit comments