【LeetCode 簡單題】115-下一個更大元素I
宣告:
今天是第115道題。給定兩個沒有重複元素的陣列 nums1
和 nums2
,其中nums1
是 nums2
的子集。找到 nums1
中每個元素在 nums2
中的下一個比其大的值。nums1
中數字 x 的下一個更大元素是指 x 在 nums2
中對應位置的右邊的第一個比 x 大的元素。如果不存在,對應位置輸出-1。以下所有程式碼經過樓主驗證都能在LeetCode上執行成功,程式碼也是借鑑別人的,在文末會附上參考的部落格連結,如果侵犯了博主的相關權益,請聯絡我刪除
(手動比心ღ( ´・ᴗ・` ))
正文
題目:給定兩個沒有重複元素的陣列 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。遍歷nums1,對每一個元素找到在nums2中的下一個的index,再判斷其索引是否合法(有沒有超過邊界),然後遍歷該index之後的元素是否有符合條件的,不符合就賦-1,程式碼如下。
執行用時: 124 ms, 在Next Greater Element I的Python3提交中擊敗了32.66% 的使用者
class Solution:
def nextGreaterElement(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
res = []
for i in nums1:
tar_i = nums1.index(i)+1
if tar_i < len(nums2):
for j in range(tar_i,len(nums2)):
if nums2[j] > i:
res.append(nums2[j])
break
elif j == len(nums2)-1:
res.append(-1)
else:
res.append(-1)
return res
解法2。先用nums2構建1個字典存放這裡面元素的大小關係,鍵是元素值,鍵值是有對應更大的下一個元素的值,其他找不到的就不會出現在字典裡,然後遍歷nums1,匹配鍵,找不到為-1,找到返回其對應鍵值,程式碼如下。
執行用時: 64 ms, 在Next Greater Element I的Python3提交中擊敗了81.48% 的使用者
class Solution:
def nextGreaterElement(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
dic = {}
stack = []
for i in nums2:
while stack and stack[-1] < i:
dic[stack.pop()] = i
stack.append(i)
return [dic.get(i,-1) for i in nums1]
結尾
解法1&解法2:https://blog.csdn.net/huhehaotechangsha/article/details/80782023