503. 下一个更大元素 II(中等)
核心思想
维护一个单调递减的单调栈(非严格)
但是由于是循环的,做两次循环即可
代码
public int[] nextGreaterElements(int[] nums) {
Deque<Integer> dq = new ArrayDeque<>();
int[] res = new int[nums.length];
Arrays.fill(res, -1);
for(int k = 0; k < 2; k++){
for(int i = 0; i < nums.length; i++){
//维护单调递减的单调栈(非严格)
if(dq.isEmpty() || nums[i] <= nums[dq.getLast()]){
dq.add(i);
}else{
// 找到答案
while(!dq.isEmpty() && nums[i] > nums[dq.getLast()]){
res[dq.getLast()] = nums[i];
dq.pollLast();
}
dq.add(i);
}
}
}
return res;
}
本文作者:Shie1d
本文链接:https://www.cnblogs.com/ganyq/p/18109108
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步