一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第9天，点击查看活动详情。

题目描述

给定一个循环数组 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路分析

• 今天的算法题目是求下一个更大元素。我们首先需要理解题目，结合示例。题目给出的是循环数组，我们首先可以定义一个2倍数组，方便查找更大的元素。然后按照要求比较。实现代码如下：

通过代码

public class DayCode {
    public static void main(String[] args) {
        int[] nums = new int[]{1, 2, 1};
        int[] ans = new DayCode().nextGreaterElements(nums);
        System.out.println("ans is " + Arrays.toString(ans));
    }

    /**
     * 链接：https://leetcode-cn.com/problems/next-greater-element-ii
     * 时间复杂度O(n * n)
     * 空间复杂度O(n)
     * @param nums
     * @return
     */
    public int[] nextGreaterElements(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];

        int[] tempNums = new int[2 * n];
        for (int i = 0; i < n; i++) {
            tempNums[i] = nums[i];
            tempNums[n + i] = nums[i];
        }

        Map<Integer, Integer> map = new HashMap<>(n);
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < 2 * n && j < n + i; j++) {
                if (tempNums[j] > tempNums[i]) {
                    map.put(i, tempNums[j]);
                    break;
                }
            }
        }

        for (int i = 0; i < n; i++) {
            ans[i] = map.getOrDefault(i, -1);
        }
        return ans;
    }
}

总结

• 今天的题目看着简单，我首先用暴力方法提交了一次，提交不通过，优化之后才明白题目的含义。暴力方法虽然时间复杂度不优，但是帮助我们理解题意还是很有帮助！
• 坚持算法每日一题，加油！