Leetcode刷題(python)之 兩個陣列的交集 II
阿新 • • 發佈:2019-02-12
給定兩個陣列,寫一個方法來計算它們的交集。
例如:
給定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
執行效率