解題思路：

因為這兩個int型的vector中不包含重復元素，所以考慮使用c++中的unordered_map，對num2進行遍歷，找到每一個元素的下一個更大元素，這樣的話，時間復雜度就是O(n)。

如果使用暴力的方法：對nums1中的每個int，在nums2中去找下一個更大元素，時間復雜度將會是O(num1.size() * nums2.size())，太過暴力。

使用一個unordered_map和一個stack進行處理的方法如下：

• 大循環遍歷vector nums2；
• 當棧不為空且棧頂元素小于nums2當前元素時，說明找到了棧頂元素的下一個更大元素，將其記錄在unordered_map中；
• 如果棧為空，或者棧頂元素仍大于當前的nums2元素，則將當前的nums2元素入棧。

循環下來之后，nums2中的一些元素可能會沒有相應的下一個更大元素，留在了棧中，但我們不管這個了，不用再將其對應-1存進unordered_map中（為了省空間）。而且，最終結果可以直接在nums1中進行更新（省空間）。

?

class Solution { public:vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {unordered_map<int, int> a;stack<int> b;//遍歷nums2for(auto x: nums2){while(!b.empty() && b.top() < x){a[b.top()] = x;b.pop();}b.push(x);}//將結果直接在nums1中進行更新for(int i = 0; i < nums1.size(); i++){if(a.count(nums1[i])) //利用count函數來看看unordered_map中是否有這個key，有的話返回1，無 則返回0nums1[i] = (a[nums1[i]]);else nums1[i] = -1;}return nums1;} };

?

?

?

總結

以上是生活随笔為你收集整理的LeetCode 496. 下一个更大元素 I 解题思路及C++实现的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。