每日一題---3

 

350. 兩個數組的交集 II

給定兩個數組，編寫一個函數來計算它們的交集。

示例 1:

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

示例 2:

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

說明：

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

進階:

• 如果給定的數組已經排好序呢？你將如何優化你的算法？
• 如果 nums1 的大小比 nums2 小很多，哪種方法更優？
• 如果 nums2 的元素存儲在磁盤上，磁盤內存是有限的，並且你不能一次加載所有的元素到內存中，你該怎麼辦？

這題唯一要注意的是取得交集是包含重複元素的，最簡單的就是暴力了，稍微優化下可以使用位圖，hashmap之類的

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        l=[]
        result=[]
        for i in nums1:
            l.append([i,min(nums1.count(i), nums2.count(i))])
        for i in l:
            if i[0] not in result:
                result += [i[0]] * i[1]
        return result

優化後 

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        inter = set(nums1) & set(nums2)
        l = []
        for i in inter:
            l += [i] * min(nums1.count(i), nums2.count(i))  
        return l