华为od统一考试B卷【阿里巴巴找黄金宝箱(IV)】

题目概括：找出下一个最大元素

代码实现：C++，Java，Python3，JS，Golang，C

贫如洗的椎夫阿里巴巴在去砍柴的路上，无意中发现了强盗集团的藏宝地，藏宝地有编号从0-N的子，每个箱子上面有一人数字，箱了排列成一个环，编号最大的箱子的下一个是编号为0的箱子。请输出每个箱了贴的数字之后的第一个比它大的数，如果不存在则输出-1。
输入描述
输入一个数字字串，数字之间使用逗号分隔，例如: 1,2,3,1
1≤ 字串中数字个数 ≤10000:
-100000≤每个数字值≤100000

输出描述
下一个大的数列表，以逗号分隔，例如: 2,3,6,-1,6
示例1：
输入

2,5,2
输出

5,-1,5
说明

第一个2的下一个更大的数是5数字5找不到下一个更大的数:第二个2的下一个最大的数需要循环搜索，结果也是 5

示例2：
输入

3,4,5,6,3
输出

4,5,6-1,4

【题目解析：】

思路及算法

我们可以使用单调栈解决本题。单调栈中保存的是下标，从栈底到栈顶的下标在数组 nums\textit{nums}nums 中对应的值是单调不升的。

每次我们移动到数组中的一个新的位置 iii，我们就将当前单调栈中所有对应值小于 nums[i]\textit{nums}[i]nums[i] 的下标弹出单调栈，这些值的下一个更大元素即为 nums[i]\textit{nums}[i]nums[i]（证明很简单：如果有更靠前的更大元素，那么这些位置将被提前弹出栈）。随后我们将位置 iii 入栈。

但是注意到只遍历一次序列是不够的，例如序列 [2,3,1][2,3,1][2,3,1]，最后单调栈中将剩余 [3,1][3,1][3,1]，其中元素 [1][1][1] 的下一个更大元素还是不知道的。

一个朴素的思想是，我们可以把这个循环数组「拉直」，即复制该序列的前 n−1n-1n−1 个元素拼接在原序列的后面。这样我们就可以将这个新序列当作普通序列，用上文的方法来处理。

而在本题中，我们不需要显性地将该循环数组「拉直」，而只需要在处理时对下标取模即可。

【C++实现：】

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n = nums.size();
        vector<int> ret(n, -1);
        stack<int> stk;
        for (int i = 0; i < n * 2 - 1; i++) {
            while (!stk.empty() && nums[stk.top()] < nums[i % n]) {
                ret[stk.top()] = nums[i % n];
                stk.pop();
            }
            stk.push(i % n);
        }
        return ret;
    }
};

【Java实现：】

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] ret = new int[n];
        Arrays.fill(ret, -1);
        Deque<Integer> stack = new LinkedList<Integer>();
        for (int i = 0; i < n * 2 - 1; i++) {
            while (!stack.isEmpty() && nums[stack.peek()] < nums[i % n]) {
                ret[stack.pop()] = nums[i % n];
            }
            stack.push(i % n);
        }
        return ret;
    }
}

【Python实现：】

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

        for i in range(n * 2 - 1):
            while stk and nums[stk[-1]] < nums[i % n]:
                ret[stk.pop()] = nums[i % n]
            stk.append(i % n)
        
        return ret

其余解法具体见：https://leetcode.cn/problems/next-greater-element-ii/solutions/637573/xia-yi-ge-geng-da-yuan-su-ii-by-leetcode-bwam/

力扣原题503，自己记录一下