LeetCode496. 下一个更大元素 I

思路一:暴力解法

遍历取出数组nums1中的每一个数,然后将每个数在nums2数组中找到下一个最大的数

详细代码:

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
		int[] nums3=new int[nums1.length];
        for(int j=0;j<nums1.length;j++){ //遍历所有的nums1
        	int num=nums1[j];
        	boolean flag=false;
        	for(int i=0;i<nums2.length;i++) {
        		if(num==nums2[i])
        			flag=true;
        		if(flag) { //找到才开始比较
        			if(nums2[i]>num) {
        				nums3[j]=nums2[i];
        				break;
        			}
        		}
        		if(flag==true&&nums3[j]==0)
        			nums3[j]=-1;
        	}
        }
        return nums3;
    }

思路二:使用栈

可以使用栈来解决,因为nums1中的元素都在nums2中,所以可以通过找nums2中的每个元素的下一个最大元素来获得一个表,最后通过查表来解决问题

首先创建栈,先遍历nums2数组,当栈不为空的时候,看栈顶元素如果比当前元素大,就将当前元素入栈,栈顶元素如果比当前元素小,就将栈顶元素弹出来,这两个就形成键值对,将键值对存到HashMap中,最后遍历nums1找到对应的数据就可以

代码:

//使用栈的思想
	public int[] nextGreaterElement(int[] nums1, int[] nums2) {
		Stack<Integer> stack=new Stack<>();//创建栈
		Map<Integer,Integer> map=new HashMap<>();
		for(int i=0;i<nums2.length;i++) {
			while(!stack.isEmpty()&&stack.peek()<nums2[i]) {
				map.put(stack.pop(),nums2[i]);
			}
			stack.push(nums2[i]);
		}
		int[] result=new int[nums1.length];
		for(int i=0;i<nums1.length;i++) {
			result[i]=map.getOrDefault(nums1[i],-1);
		}
		return result;
	}

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值