题目:
思路:我的做法是现在nums2中找到nums1的元素,然后依次往后遍历,找到第一个比该元素大的数,存起来,查nums1中的下一个元素。
代码:
#include <iostream>
#include <vector>
#include <array>
using namespace std;
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {//我第一次写的时候把它的两个参数搞反了,然后就重命名了1和2
vector<int> res;
for(int i : nums1){
int flag = -1;
for(int j = 0; j < nums2.size(); j++){
if(nums2[j] == i){
while(j < nums2.size()){
if(nums2[j] > i){
flag = nums2[j];
break;
}
j++;
}
res.push_back(flag);
break;
}
}
}
return res;
}
};
int main(){
Solution so;
int a[3] = {4,1,2};
vector<int> nums1 (a,a+3);
int b[4] = {1,3,4,2};
vector<int> nums2(b,b+4);
vector<int> res;
res = so.nextGreaterElement(nums1, nums2);
for(int i = 0; i < res.size() ; i++)
cout << res[i] << endl;
system("pause");
return 0;
}