1. 程式人生 > >Leetcode刷題(python)之 兩個陣列的交集 II

Leetcode刷題(python)之 兩個陣列的交集 II

給定兩個陣列,寫一個方法來計算它們的交集。

例如:
給定nums1 = [1, 2, 2, 1]nums2 = [2, 2], 返回 [2, 2].

注意:

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

跟進:

  • 如果給定的陣列已經排好序呢?你將如何優化你的演算法?
  • 如果 nums1 的大小比 nums2 小很多,哪種方法更優?
  • 如果nums2的元素儲存在磁碟上,記憶體是有限的,你不能一次載入所有的元素到記憶體中,你該怎麼辦?

class Solution:
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums1.sort()
        nums2.sort()
        l1 = len(nums1)
        l2 = len(nums2)
        
        result=[]
        i = 0
        j = 0
        while i<l1 and j<l2:
            if nums1[i]<nums2[j]:
                i+=1
            elif nums1[i]>nums2[j]:
                j+=1
            else:
                result.append(nums1[i])
                i+=1
                j+=1
        
        return result

執行效率