題目

給定兩個沒有重復(fù)元素的數(shù)組 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下一個比其大的值。

nums1 中數(shù)字 x 的下一個更大元素是指 x 在 nums2 中對應(yīng)位置的右邊的第一個比 x 大的元素。如果不存在，對應(yīng)位置輸出-1。

示例 1:

輸入: nums1 = [4,1,2], nums2 = [1,3,4,2].
輸出: [-1,3,-1]
解釋:
對于num1中的數(shù)字4，你無法在第二個數(shù)組中找到下一個更大的數(shù)字，因此輸出 -1。
對于num1中的數(shù)字1，第二個數(shù)組中數(shù)字1右邊的下一個較大數(shù)字是 3。
對于num1中的數(shù)字2，第二個數(shù)組中沒有下一個更大的數(shù)字，因此輸出 -1。

思路

對于數(shù)組num2，利用棧找出每一個元素的下一個最大元素，將結(jié)果保存在哈希表中，（該數(shù)&下一個更大的元素的鍵值對），之后再遍歷num1,如果hash表中沒有當前元素，表示沒有更大的元素，將-1push進結(jié)果數(shù)組，如果存在，就hash表中保存的下一個更大的數(shù)保存進結(jié)果數(shù)組。

其實mum1只是給出了部分打亂的值，相應(yīng)的輸出他們右邊第一個比它們大的元素，其實并不影響這道題目的本質(zhì)。

代碼

class Solution { public:vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {vector<int> ans;stack<int> st;unordered_map<int,int> mp;for(int i=0;i<nums.size();i++){while(!st.empty() && st.top()<nums[i]){mp[st.top()]=nums[i];st.pop();}st.push(nums[i]);}for(int i=0;i<findNums.size();i++){if(mp.count(findNums[i])==0) ans.push_back(-1);else ans.push_back(mp[findNums[i]]);}return ans;} };

轉(zhuǎn)載于:https://www.cnblogs.com/Bipolard/p/10173430.html

總結(jié)

以上是生活随笔為你收集整理的下一个更大元素 I(LeetCode 496)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。