1. 程式人生 > >350. 兩個陣列的交集 II

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 的元素儲存在磁碟上,磁碟記憶體是有限的,並且你不能一次載入所有的元素到記憶體中,你該怎麼辦?
class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        rs = []
        flag = []
        for i,val_i in enumerate(nums1):
            for j,val_j in enumerate(nums2):
                if (val_i==val_j) and (j not in flag):
                    flag.append(j)
                    rs.append(val_i)
                    break
        return rs



def main():
    a = [1,2,2,1]
    b = [2,2]
    myresult = Solution()
    print(myresult.intersect(a,b))

if __name__ == "__main__":
    main()

 別人的: