• 一、題目
• 二、解法
• 結語

一、題目

給定兩個陣列，編寫一個函式來計算它們的交集。

例1：

輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2,2]

例2：

輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [4,9]

說明：

• 輸出結果中每個元素出現的次數，應與元素在兩個陣列中出現的次數一致
• 我們可以不考慮輸出結果的順序

二、解法

首先把兩個陣列都排序，然後兩個陣列進行遍歷比較，
當值相等時，兩個陣列都往後移動一位，並且相等的那個值加入到 result 陣列中
當值不相等時，值小的陣列往後移動一位(因為我們已經排序過了，如果是值大的陣列往後移動一位，那麼肯定還是比值小的要大)

程式碼如下：

nums1.sort()
nums2.sort()
result = []
i, j = 0, 0
while i<len(nums1) and j<len(nums2):
    if nums1[i]==nums2[j]:
        result.append(nums1[i])
        i += 1
        j += 1
    elif nums1[i]>nums2[j]:
        j += 1
    elif nums1[i]<nums2[j]:
        i += 1
return result

時間 52ms，擊敗了 93%

結語

歡迎關注我的公眾號 瘋子的Python筆記