关于栈的学习-05

[leetcode] 496:下一个更大元素I

问题描述

给定两个无重复元素的数组nums1与nums2，nums1 是 nums2 的子集。要求找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
示例

1）输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
   输出: [-1,3,-1]
2）输入: nums1 = [2,4], nums2 = [1,2,3,4].
   输出: [3,-1]
3）输入: nums1 = [], nums2 = [1,2,3,4].
   输出: []

求解思路

采用枚举法加中间变量标记的方式对该问题进行了求解
java程序如下

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int[] num = new int[nums1.length];
        if(nums1.length<1 || nums2.length<1){
            return nums1;
        }
        for(int i=0;i<nums1.length;i++){
            /**寻找nums1中的元素及其比他大的元素*/
            int mid =1;
            int mid0 = 1;
            for(int j=0;j<nums2.length;j++){
                if(nums1[i]==nums2[j]){
                    mid = 0;
                }
                if (mid<1 && mid0>0 && nums1[i]<nums2[j]){
                    mid0 = 0;
                    num[i]=nums2[j];
                }
            }
            if (mid0>=1){
                num[i]=-1;
            }
        }  
        return num;
    }
}

总结

尝试用栈的数据结构进行求解，结果显示超出时间限制。
解题思路如下：定义两个堆栈S1与S2，将数组nums2逆向压入栈S1中，依次遍历nums1中的元素，将不满足约束（1.栈顶(S1)未出现nums1中的元素；2.栈顶元素值小于等于nums1中的元素值）的元素从S1中出栈，压入S2中；将满足约束的元素存入数组num中；S1为空时，未找到一个更大的元素，则将-1存入数组中；再将S2中的元素依次出栈，压入S1中，继续执行对nums1遍历。最后输出num。