leetcode496-Next Greater Element I

博客围绕LeetCode题目展开,给定无重复元素的数组nums1和nums2,需找出nums1中每个元素在nums2对应位置右侧第一个比其大的元素。分析了两种解法,一是两次for循环遍历nums2,二是利用栈存储不确定元素,避免两次遍历。

题目:

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
给你两个 没有重复元素 的数组 nums1 和 nums2 ,下标从 0 开始计数,其中nums1 是 nums2 的子集。
对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1 。
返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素 。

分析:

这道题目最直接的思路就是俩次for循环遍历nums2,找到第一个比当前元素大的值,存入map,然后遍历nums1即可
还有一种办法,可以不用俩次for循环遍历nums2,我们想一下俩次遍历nums2的目的,就是因为有可能比当前元素最大的那个值还在后面,在当下的遍历没办法确定下来,这种时候就要想到一个数据结构即栈,栈可以先暂时把不确定的元素存起来

import java.util.Map;
import java.util.HashMap;
import java.util.Deque;
import java.util.LinkedList;

public class nextGreaterElementI {
	public static void main(String[] args) {
		int[] arr = {4,1,2};
		int[] brr = {1,3,4,2};
		int[] crr = getNextMax(arr,brr);
		for(int i = 0;i<crr.length;i++) {
			System.out.println(crr[i]);
		}
	}
	public static int[] getNextMax(int[] arr,int[] brr) {
		Map mp = new HashMap();
		Deque stack = new LinkedList();
		for(int i = 0;i<brr.length;i++) {
			while(stack.size() != 0 && (int) stack.peek() < brr[i]) {
				mp.put(stack.peek(),brr[i]);
				stack.pop();
			}
			stack.push(brr[i]);
		}
		int[] crr = new int[arr.length];
		int j = 0;
		for(int i = 0;i<arr.length;i++) {
			if(mp.containsKey(arr[i])) {
				crr[j++] = (int) mp.get(arr[i]);
			} else {
				crr[j++] = -1;
			}
		}
		return crr;
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值