【LeetCode】496. Next Greater Element I

题目：

思路：我的做法是现在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;  
}