1. 程式人生 > >【LeetCode 簡單題】115-下一個更大元素I

【LeetCode 簡單題】115-下一個更大元素I

宣告:

今天是第115道題。給定兩個沒有重複元素的陣列 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下一個比其大的值。nums1 中數字 x 的下一個更大元素是指 x 在 nums2 中對應位置的右邊的第一個比 大的元素。如果不存在,對應位置輸出-1。以下所有程式碼經過樓主驗證都能在LeetCode上執行成功,程式碼也是借鑑別人的,在文末會附上參考的部落格連結,如果侵犯了博主的相關權益,請聯絡我刪除

(手動比心ღ( ´・ᴗ・` ))

正文

題目:給定兩個沒有重複元素的陣列 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下一個比其大的值。nums1 中數字 x 的下一個更大元素是指 x 在 nums2 中對應位置的右邊的第一個比 大的元素。如果不存在,對應位置輸出-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。

注意:

  1. nums1nums2中所有元素是唯一的。
  2. nums1nums2 的陣列大小都不超過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