503——下一个更大元素II

题目描述

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

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

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

题解

1. 参考官方题解
2. C语言实现

int* nextGreaterElements(int* nums, int numsSize, int* returnSize){
    
    
    *returnSize=numsSize;
    int* next=(int *)malloc(numsSize*sizeof(int));
    int* stack=(int *)malloc(numsSize*sizeof(int));//栈
    int top=-1;//栈指针

    for(int i=0;i<numsSize;i++)
    {
    
    
        while(top>-1&&nums[stack[top]]<nums[i])//当前元素比栈顶元素大则栈顶元素出栈
        {
    
    
            next[stack[top--]]=nums[i];
        }
        stack[++top]=i;//当前元素入栈
    }
    //循环查找
    for(int i=0;top>-1&&i<stack[top];i++)
    {
    
    
        while(top>-1&&nums[stack[top]]<nums[i])//当前元素比栈顶元素大则栈顶元素出栈
        {
    
    
            next[stack[top--]]=nums[i];
        }
    }
    //剩下无下一个更大元素
    while(top>-1)
    {
    
    
        next[stack[top--]]=-1;
    }
    free(stack);
    return next;
}

3. 大致思路：使用栈存储数组的下标，将栈顶的下标对应的元素与当前元素比较。栈中存储下标对应的元素大小单调不增。