给定两个没有重复元素的数组 nums1
和 nums2
,其中nums1
是 nums2
的子集。找到 nums1
中每个元素在 nums2
中的下一个比其大的值。
nums1
中数字 x 的下一个更大元素是指 x 在 nums2
中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。
示例 1:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2]. 输出: [-1,3,-1] 解释: 对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。 对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。 对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。
示例 2:
输入: nums1 = [2,4], nums2 = [1,2,3,4]. 输出: [3,-1] 解释: 对于num1中的数字2,第二个数组中的下一个较大数字是3。 对于num1中的数字4,第二个数组中没有下一个更大的数字,因此输出 -1。
注意:
nums1
和nums2
中所有元素是唯一的。nums1
和nums2
的数组大小都不超过1000。
暴力解
1 class Solution { 2 public: 3 vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) { 4 for(int i = 0; i < findNums.size(); i++){ 5 for(int j = 0; j < nums.size(); j++){ 6 bool flag = false; 7 if(nums[j] == findNums[i]){//找到findNums[i]在nums中的位置 8 flag = true; 9 bool f = true; 10 for(int k = j; k < nums.size(); k++){//找nums[k]之后是否存在比nums[j]大的数 11 if(findNums[i] < nums[k]){ 12 findNums[i] = nums[k]; 13 f = false; 14 break; 15 } 16 } 17 if(f) findNums[i] = -1; //f=true表示没有找到比findNums[i]的值 18 } 19 if(flag) break; 20 } 21 } 22 return findNums; 23 } 24 };
再对nums查找之前,可以对nums进行一定的筛选;
如果nums[i]的右边没有比nums[i] 大的数, 则忽略他
用map来解,更加简便map<key, value> 这里的key是int型的表示findNums的值, value也是int型的表示比findNums大的值
1 class Solution { 2 public: 3 vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) { 4 stack<int> s; 5 map<int, int> m; 6 for(int i = 0; i < nums.size(); i++){//nums中存在比nums[i]大的数放入m中 7 while(!s.empty() && s.top() < nums[i]){ 8 m[s.top()] = nums[i]; 9 s.pop(); 10 } 11 s.push(nums[i]); 12 } 13 for(int i = 0; i < findNums.size(); i++)//如果m中存在findNums则表示nums中存在比findNums大的数 14 findNums[i] = m.count(findNums[i])? m[findNums[i]] : -1; 15 return findNums; 16 } 17 };
总结:对于在数组中找比当前元素大或者小的值的问题, 通用方法就是用stack,把没有找到符合条件的nums[i],待找到符合条件的值之后,把这些符合条件的情况全部弹出
stack,map的原理都知道,但是运用起来还是不熟练,代码量还是太少,继续累积,学习,总结