这个题,开始我是想用find和upper_bound两个函数解决,但是
upper_bound并不对。
upper_bound返回的是插入的数字的上界。
如一个数组number序列1,2,2,4.upper_bound(2)后,返回的位置是3(下标)也就是4所在的位置。
并且用的是二分查找,所以序列要求有序。
那就自己写了个循环。
还有find的返回值,找不到返回end。
最后一个返回end-1.
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
int l1=findNums.size();
int l2=nums.size();
vector<int> ans;
vector<int>::iterator it,is;
for(int i=0;i<l1;i++){
it=find(nums.begin(),nums.end(),findNums[i]);
if(it==nums.end()-1){ ans.push_back(-1); continue; }
int k=0;
for(;it!=nums.end();it++){
if(*it>findNums[i]){ ans.push_back(*it); k=1; break; }
}
if(!k) ans.push_back(-1);
}
return ans;
}
};