leetcode 497 栈

141 阅读1分钟
//给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。 
//
// 请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。 
//
// nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。 
//
// 
//
// 示例 1: 
//
// 
//输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
//输出: [-1,3,-1]
//解释:
//    对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。
//    对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。
//    对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。 
//
// 示例 2: 
//
// 
//输入: nums1 = [2,4], nums2 = [1,2,3,4].
//输出: [3,-1]
//解释:
//    对于 num1 中的数字 2 ,第二个数组中的下一个较大数字是 3 。
//    对于 num1 中的数字 4 ,第二个数组中没有下一个更大的数字,因此输出 -1 。
// 
//
// 
//
// 提示: 
//
// 
// 1 <= nums1.length <= nums2.length <= 1000 
// 0 <= nums1[i], nums2[i] <= 104 
// nums1和nums2中所有整数 互不相同 
// nums1 中的所有整数同样出现在 nums2 中 
// 
//
// 
//
// 进阶:你可以设计一个时间复杂度为 O(nums1.length + nums2.length) 的解决方案吗? 
// Related Topics 栈 
// 👍 358 👎 0
import sun.util.locale.provider.FallbackLocaleProviderAdapter;

import java.util.Stack;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int [] res = new int[nums1.length];
        Stack<Integer> stack = new Stack<>();
        for(int num: nums2){
            stack.push(num);
        }

        for(int i = 0;i < nums1.length;i++){
            Stack<Integer> temp = new Stack<>();
            boolean isFound = false;
            int cur = nums1[i];
            int max = -1;

            while(!stack.isEmpty() && !isFound){
                int top = stack.pop();
                if (top>cur){
                    max = top;
                } else if (top == cur){
                    isFound = true;
                }
                temp.push(top);

            }
            res[i] = max;
            while(!temp.isEmpty()){
                stack.push(temp.pop());
            }

        }
        return res;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

num1 是 num2的子集。 生成一个num1长度的数组 把num2全部push到stack栈 遍历num1的元素,并且从nums2的stack栈pop数字来比较,默认max=-1;如果比当前num1元素大,则max=top,如果等于则返回找到,同时把数字Push到temp栈,再pop填回原有的的栈。

for i in num1:
    max = 1
    cur = i 
    while(!stack.isEmpty() && !isFound):
    	top  = stack.pop()
    	if top > max:
       		max = top 
    	else (top == cur):
       		isFound = true
     
    temp.push(top);
    
	res[i] = max;

	while(!temp.isEmpty()):
     	stack.push(temp.pop())


return res