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] 是如上所述的 下一个更大元素 。
#include <bits/stdc++.h>
using namespace std;
int a[123141], b[21341], n, m, c[2425];
int main(){
cin >> n >> m;
for(int i = 0; i < n; i++){
cin >> a[i];
}
for(int i = 0; i < m; i++){
cin >> b[i];
}
for(int i = 0; i < n; i++)
c[i] = -1;
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
if(a[i] == b[j]){
for(int k = j + 1; k < m; k++){
if(b[k] > b[j]){
c[i] = b[k];
break;
}
}
}
}
}
for(int i = 0; i < n; i++){
cout << c[i] << " ";
}
return 0;
}
/*
3 4
4 1 2
1 3 4 2
2 4
2 4
1 2 3 4
*/