503. 下一个更大元素 II
给定一个循环数组 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]
题解:
class Solution {
public int[] nextGreaterElements(int[] nums) {
int len=nums.length;//获得数组的长度
int[] array=new int[len];
int j=0;
for(int i=0;i<len;i++){
if(i!=len-1){//因为是循环数组,当数组不等于len-1时,可以把他当作普通的数组
for(j=i+1;j<len;j++){
if(nums[i]<nums[j]){//如果后一项比前一项大就满足条件
array[i]=nums[j];
break;
}else if (j==i){//再循环完一圈之后,如果没有找到比前一项大的
array[i]=-1;
break;
}
if(j==len-1){
j=-1;//循环数组,-1是因为for时会加一
}
}
}else {//如果等于i=len-1,j直接从0开始
for(j=0;j<len;j++){
if(nums[i]<nums[j]){
array[i]=nums[j];
break;
}else if (j==i){
array[i]=-1;
break;
}
if(j==len-1){
j=-1;
}
}
}
}
return array;
}
}
以上是博主的暴力解法,不值得大家提倡,奈何博主实在是菜的抠脚。。
下面是官方的题解:
方法:单调栈 + 循环数组
思路及算法
我们可以使用单调栈解决本题。单调栈中保存的是下标,从栈底到栈顶的下标在数组 nums 中对应的值是单调不升的。
每次我们移动到数组中的一个新的位置 ii,我们就将当前单调栈中所有对应值小于nums[i] 的下标弹出单调栈,这些值的下一个更大元素即为nums[i](证明很简单:如果有更靠前的更大元素,那么这些位置将被提前弹出栈)。随后我们将位置 i 入栈。
但是注意到只遍历一次序列是不够的,例如序列 [2,3,1],最后单调栈中将剩余 [3,1],其中元素 [1] 的下一个更大元素还是不知道的。
一个朴素的思想是,我们可以把这个循环数组「拉直」,即复制该序列的前 n−1 个元素拼接在原序列的后面。这样我们就可以将这个新序列当作普通序列,用上文的方法来处理。
而在本题中,我们不需要显性地将该循环数组「拉直」,而只需要在处理时对下标取模即可。
代码:
class Solution {
public int[] nextGreaterElements(int[] nums) {
int n = nums.length;
int[] ret = new int[n];
Arrays.fill(ret, -1);
Deque<Integer> stack = new LinkedList<Integer>();
for (int i = 0; i < n * 2 - 1; i++) {
while (!stack.isEmpty() && nums[stack.peek()] < nums[i % n]) {
ret[stack.pop()] = nums[i % n];
}
stack.push(i % n);
}
return ret;
}
}