【栈】Leetcode 496 下一个更大元素I
- 解法1 两个单调栈
- 解法2
---------------🎈🎈题目链接🎈🎈-------------------
解法1 两个单调栈
两个栈进行操作,一个栈用来遍历寻找,一个栈用来保留
将nums2中的元素入栈,之后遍历nums1,
如果栈顶元素大于nums1[i],则记录max
如果栈顶元素小于nums1[i],则弹出栈顶元素至
创建栈:Stack<Integer> mystack = new Stack<>();
栈顶元素:mystack.peek();
栈顶元素弹出:mystack.pop();
栈是否为空:mystack.isEmpty();
加入栈:mystack.push();
时间复杂度O(N)
空间复杂度O(N)
class Solution {public int[] nextGreaterElement(int[] nums1, int[] nums2) {int[] result = new int[nums1.length];Stack<Integer> mystack = new Stack<>();Stack<Integer> tempstack = new Stack<>();for(int i = 0; i<nums2.length; i++){mystack.push(nums2[i]);}for(int i = 0; i <nums1.length; i++){boolean sig = true;int max = -1;while(sig && !mystack.isEmpty()){if(nums1[i] < mystack.peek()){max = mystack.peek();}else if(nums1[i] == mystack.peek()){sig = false;while(!tempstack.isEmpty()){mystack.push(tempstack.pop());}continue;}tempstack.push(mystack.pop());}result[i] = max;}return result;}
}
解法2
时间复杂度O(N)
空间复杂度O(N)