大致题意为要求找到nums1中x在nums2中对应位置右侧并且比x大的元素,有该元素时要求输出该元素,没有符合条件的元素或者越界时输出-1,其中nums1是nums2的子集,最后的结果要返回一个和nums1长度相同的数组
我们要明确题目要求我们干什么,无非就是两件事
1.找到元素的位置
2.输出该元素右侧且比下大的元素
所以我们需要遍历nums2找到nums1中的对应元素,通过两个while循环找到对应元素右侧是否有比该元素大的更大元素,同时我们也要考虑到数组的长度,越界时的情况也要考虑到,在编写代码时还要创建一个数组用来返回结果
代码如下
class Solution {
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size();
int n = nums2.size();//定义数组长度防止越界
vector<int>res(m);//创建输出结果的数组
for (int i = 0; i < m; ++i) {//遍历nums1
int j = 0;
while (j < n && nums2[j] != nums1[i]) {
++j;//在nums2中找nums1的对应元素
}
int k = j + 1;
while (k < n && nums2[k] < nums2[j]) {
++k;//向右找更大的元素
}
res[i] = k < n ? nums2[k] : -1;//判定越界时输出-1
}
return res;
}
};
本人编程小白一枚,如有错误,欢迎各位大佬指教~