leetcode：503. 下一个更大元素 II

503. 下一个更大元素 II

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

示例 1:

输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数；
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
注意: 输入数组的长度不会超过 10000。

思路

假设不是循环数组，是一个普通数组。

• 满足递归式：

// 设这个数组为a
next_bigger[i] = a[i+1]
// 如果a[i+1]>a[i],那么a[i+1]肯定就是a[i]的下一个更大的元素
// 否则，如果a[i+1]的下一个更大的元素比a[i]大，那么a[i+1]的下一个更大的元素就是a[i]的下一个更大的元素
// 。。。一直循环
while(a[i]>=next_bigger[i]) 
	next_bigger[i] = next_bigger[（next_bigger[i]）在数组a中的下标];

不难理解，如果a[i+1]<=a[i]，那么比a[i+1]还小的数，就不用考虑了，可以直接跳到a[i+1]的下一个更大元素。这里我们就从后往前遍历，不断更新next_bigger数组就行了。

但是题目是循环数组怎么解决呢？

只需要重复两次遍历就行啦。

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int v1[10000];	// next_bigger
        int v2[10000];	// 记录next_bigger在a中的下标
        for(int i = nums.size()-1; i >= 0; i --){
            if(i == nums.size()-1) {
                v1[i] = -1;
                v2[i] = -1;
                continue;
            }
            if(nums[i] < nums[i+1]){
                v1[i] = nums[i+1];
                v2[i] = i+1;
            } 
            else {
                int k = v2[i+1];
                while(k >= 0&&nums[i] >= nums[k])
                    k = v2[k];
                if(k == -1) {
                    v1[i] = -1;
                    v2[i] = -1;
                }
                else {
                    v1[i] = nums[k];
                    v2[i] = k;
                }
            }
        }
        
        for(int i = nums.size()-1; i >= 0; i --){
            int n = (i+1>=nums.size())?i+1-nums.size():i+1;
            if(nums[i] < nums[n]){
                v1[i] = nums[n];
                v2[i] = n;
            } 
            else {
                int k = v2[n];
                while(k >= 0&&nums[i] >= nums[k])
                    k = v2[k];
                if(k == -1) {
                    v1[i] = -1;
                    v2[i] = -1;
                }
                else {
                    v1[i] = nums[k];
                    v2[i] = k;
                }
            }
        }
        vector<int> v;
        for(int i = 0; i < nums.size(); i ++)
            v.push_back(v1[i]);
        return v;
    }
};