題目描述:
給定兩個沒有重複元素的數組 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。
思路描述:
思路一:利用棧去將nums1中滿足小於相鄰右邊元素的元素存儲在hash表中,key是該元素的值、value是該元素右邊的相鄰元素。從而利用哈希查找來實現輸出。
思路二:不使用棧,直接利用哈希表,key是nums1中的值,value則是nums1中值對應的地址ID。
實現代碼一:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
stack<int> s;
unordered_map<int, int> m;
for (int n : nums) {
while (s.size() && s.top() < n) {
m[s.top()] = n;
s.pop();
}
s.push(n);
}
vector<int> ans;
for (int n : findNums)
ans.push_back(m.count(n) ? m[n] : -1);
return ans;
}
};
實現代碼二:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
int findNumsSize = findNums.size();
int numsSize = nums.size();
// int aa=sizeof(findNums);
// int b=sizeof(nums);
int* a = new int[findNumsSize];
vector<int> ans;
int i;
int j;
int k;
for (i=0;i<findNumsSize;i=i+1){
a[i]=findNums[i];
for (j=0;j<numsSize;j=j+1)
if(findNums[i]==nums[j]){
k=j+1;
break;
}
for(k=j+1;k<numsSize;k=k+1)
if(nums[k]>a[i]){
a[i]=nums[k];
break;
}
if(a[i]==findNums[i])
a[i]=-1;
}
for (i=0;i<findNumsSize;i++){
ans.push_back(a[i]);
}
return ans;
}
};
實現代碼三:
class Solution {
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {
vector<int> result;
stack<int> s1;
for (int i=0;i<findNums.size();i++){
for (int j=(nums.size()-1);j>=0;j--){
if (findNums[i]<nums[j]){
s1.push(nums[j]);
}
else if (findNums[i]==nums[j])
break;
else
continue;
}
if(s1.empty())
result.push_back(-1);
else
result.push_back(s1.top());
while(!s1.empty())
s1.pop();
}
return result;
}
};
實現代碼四
class Solution{
public:
vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums){
std::size_t len = nums.size();
vector<int> ret;
unordered_map<int,int> mp;
for(int i = 0;i < len; ++i){
mp[nums[i]] = i;
}
for(int i = 0; i < findNums.size(); ++i){
int index = mp[findNums[i]] + 1;
if(index < len && nums[index] > findNums[i])
ret.push_back(nums[index]);
else
ret.push_back(-1);
}
return ret;
}
};
關於此題其實還有其他的解法、但是個人認爲利用hash查找來實現算是比較好的思路,歡迎指正!