方法:单调栈和循环数组
单调栈:从栈底到栈顶单调递增或者单调递减的栈
循环数组:数组的首尾相连(逻辑上的)可以通过求余形成逻辑上的循环数组
注:题目给的数组是普通数组,ans是答案数组,利用求余可以得到逻辑上的循环数组,栈也可以用数组来代替,只是赋值和删除方式有所约束即可。
循环的次数就是原数组的2倍数-1,因为走一遍不能保证找完数组中每一个数的的下一个更大的数,比如最后的几个数的下一个更大的数可能就在它的前面,如下图:
int* nextGreaterElements(int* nums, int numsSize, int* returnSize) {
*returnSize = numsSize;
if (numsSize == 0) {
return NULL;
}
int* ret = malloc(sizeof(int) * numsSize); // 开辟空间
memset(ret, -1, sizeof(int) * numsSize); // 都初始化为-1
int stk[numsSize * 2 - 1], top = 0; // 定义栈
for (int i = 0; i < numsSize * 2 - 1; i++) {
while (top > 0 && nums[stk[top - 1]] < nums[i % numsSize]) {
ret[stk[top - 1]] = nums[i % numsSize]; // 答案数组的赋值也是出栈过程看图即可
top--; // 出栈
}
stk[top++] = i % numsSize; // 为了实现起来方便栈中存储的是下标而不是数值
}
return ret; // 返回数组
}
作者:LeetCode-Solution
链接:https://leetcode.cn/problems/next-greater-element-ii/solution/xia-yi-ge-geng-da-yuan-su-ii-by-leetcode-bwam/
来源:力扣(LeetCode)
代码参考如上