LeetCode: 496 Next Greater Element I(easy)
题目:
You are given two arrays (without duplicates) nums1
and nums2
where nums1
’s elements are subset of nums2
. Find all the next greater numbers for nums1
's elements in the corresponding places of nums2
.
The Next Greater Number of a number x in nums1
is the first greater number to its right in nums2
. If it does not exist, output -1 for this number.
Example 1:
Input: nums1 = [4,1,2], nums2 = [1,3,4,2].
Output: [-1,3,-1]
Explanation:
For number 4 in the first array, you cannot find the next greater number for it in the second array, so output -1.
For number 1 in the first array, the next greater number for it in the second array is 3.
For number 2 in the first array, there is no next greater number for it in the second array, so output -1.
Example 2:
Input: nums1 = [2,4], nums2 = [1,2,3,4].
Output: [3,-1]
Explanation:
For number 2 in the first array, the next greater number for it in the second array is 3.
For number 4 in the first array, there is no next greater number for it in the second array, so output -1.
Note:
- All elements in
nums1
andnums2
are unique. - The length of both
nums1
andnums2
would not exceed 1000.
代码:
自己的:
1 class Solution { 2 public: 3 vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) { 4 vector<int> result; 5 int s1 = findNums.size(); 6 int s2 = nums.size(); 7 bool b = 0; 8 for (int i=0; i<s1; i++){ 9 int tem = -1; 10 for (int j=0; j <s2; j++){ 11 if (findNums[i] == nums[j]){ 12 for (int k=j; k<s2; k++){ 13 if (nums[k]>findNums[i]){ 14 tem = nums[k]; 15 break; 16 } 17 } 18 } 19 } 20 result.push_back(tem); 21 } 22 return result; 23 } 24 };
别人的:
1 class Solution { 2 public: 3 vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) { 4 stack<int> s; 5 unordered_map<int,int> hash; 6 for(int i=0;i<nums.size();i++){ 7 if(s.empty()){ 8 s.push(nums[i]); 9 } 10 else if(nums[i] > s.top()){ 11 while(!s.empty() && s.top()<nums[i]){ 12 hash[s.top()] = nums[i]; 13 s.pop(); 14 } 15 s.push(nums[i]); 16 } 17 else s.push(nums[i]); 18 } 19 while(!s.empty()){ 20 hash[s.top()] = -1; 21 s.pop(); 22 } 23 vector<int> res; 24 for(int i=0;i<findNums.size();i++){ 25 res.push_back(hash[findNums[i]]); 26 } 27 return res; 28 } 29 };
unordered_map类是c++11标准的内容,具体介绍见链接:https://msdn.microsoft.com/zh-cn/library/bb982522.aspx