题目
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
题解
题解一:
- 使用双重循环
- 第二层循环从j = 1开始,每次取(i+j)%nums[nums.Length]
- 时间复杂度为O(n²),空间复杂度为O(n)
- 代码
public int[] NextGreaterElements(int[] nums) {
int len = nums.Length;
int[] res = new int[len];
for (int i = 0; i < len; i++)
{
res[i] = -1;
for (int j = 1; j < len; j++)
{
if (nums[(j+i) % len] > nums[i])
{
res[i] = nums[(j+i) % len];
Console.WriteLine(res[i]);
break;
}
}
}
return res;
}
题解二:
- 使用单调栈
- 假设存在一个[1,2,1,1,2,1]改造数组,这样一次循环的数组
- 维持一个Stack记录元素的index,Stack中的index对应的元素单调递减,
- 从改造数组的最后一个元素向前扫描
- while当前元素比Stack的顶部元素大or Stack为空:
Stack.Pop
Stack为空?当前元素的结果是-1:Stack.Peek - 时间复杂度O(n),空间复杂度O(n)
- 代码
public int[] NextGreaterElements(int[] nums) {
int len = nums.Length;
int[] res = new int[len];
Stack<int> stack = new Stack<int>();
for (int i = len*2-1; i >= 0; --i)
{
while (stack.Count>0&& nums[stack.Peek()]<=nums[i%len])
{
stack.Pop();
}
res[i % len] = stack.Count > 0 ? nums[stack.Peek()] : -1;
stack.Push(i % len );
}
return res;
}