下一个更大元素1


下一个更大元素1

题目

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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] 是如上所述的 下一个更大元素

示例 1:
输入:nums1 = [4,1,2], nums2 = [1,3,4,2].
输出:[-1,3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 4 ,用加粗斜体标识,nums2 = [1,3,4,2] 不存在下一个更大元素,所以答案是 -1
- 1 ,用加粗斜体标识,nums2 = [1,3,4,2] 下一个更大元素是 3
- 2 ,用加粗斜体标识,nums2 = [1,3,4,2] 不存在下一个更大元素,所以答案是 -1

示例 2:
输入:nums1 = [2,4], nums2 = [1,2,3,4].
输出:[3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 2 ,用加粗斜体标识,nums2 = [1,2,3,4] 下一个更大元素是 3
- 4 ,用加粗斜体标识,nums2 = [1,2,3,4] 不存在下一个更大元素,所以答案是 -1

解法1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
思路:
计算nums2每个数字的下一个更大数,生成数字->大数的map,遍历num1用map生成最终的结果

代码:
/**
* @param Integer[] $nums1
* @param Integer[] $nums2
*
* @return Integer[]
*/
function nextGreaterElement($nums1, $nums2)
{
$count = count($nums2);

$res = [];
$l = [];

// 从末尾开始遍历
for ($i = $count - 1; $i >= 0; $i--) {
$num = $nums2[$i];
// 设置默认值为-1
$res[$num] = -1;

// 栈中必须有数据才开始循环,没数据默认为-1
while ($l) {
if ($num > end($l)) {
// 如果当前值大于栈中的末尾值,证明末尾值不是当前值的下一个大数
// 从栈的末尾移除
// 当前值比末尾值大,所有当前值必定是接下来遍历值能对比的大数
// 比如[1,5,3,8,10], 栈的结果是[10,8,5,1], 从大到小排序
array_pop($l);
} else {
// 记录下一个大数的值与当前值对应的map
$res[$num] = end($l);
break;
}
}

//入栈
$l[] = $num;
}

$d = [];
foreach ($nums1 as $item) {
// res是值对应的大数map
$d[] = $res[$item];
}

return $d;
}

解法2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
思路:
获取nums2中数字对应的下标位置的map,
遍历nums1,在其中遍历nums2,下标从map里获取

代码:
/**
* @param Integer[] $nums1
* @param Integer[] $nums2
*
* @return Integer[]
*/
function nextGreaterElement($nums1, $nums2)
{
$count = count($nums2);

// 生成nums2->index的map
$map = array_flip($nums2);

$res = [];
foreach ($nums1 as $item) {
// 获取nums1数字在nums2中的下标
$index = $map[$item];

// 设置默认值-1
$num = -1;
for ($i = $index + 1; $i < $count; $i++) {
if ($nums2[$i] > $item) {
// 遍历nums2直到获取比nums1中大的数字
$num = $nums2[$i];
break;
}
}
// 记录结果
$res[] = $num;
}

return $res;
}