496. 下一个更大元素 I
程序员文章站
2022-03-22 16:25:50
...
下一个更大元素 I
给定两个没有重复元素的数组 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。
思路+代码+注释:
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
/*
思路:使用map记录nums2中每个元素及其下标索引,然后遍历nums1中的元素在map中找到对应的下标,然后从该下标后开始查找比它大的元素,如果有进行记录,
没有就记录-1
*/
int[] res=new int[nums1.length];
HashMap<Integer,Integer> map=new HashMap<>();
for (int i = 0; i < nums2.length; i++) {
map.put(nums2[i],i);
}
for (int i = 0; i < nums1.length; i++) {
boolean flag=true;
int num=nums1[i];
int index=map.get(num);
for (int j = index+1; j < nums2.length; j++) {
if (nums2[j]>num)
{
flag=false;
res[i]=nums2[j];
break;
}
}
if (flag)
{
res[i]=-1;
}
}
return res;
}