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

leetcode 兩個陣列的交集

leetcode 兩個陣列的交集

  • 題目
    給定兩個陣列,編寫一個函式來計算它們的交集。
  • 說明
    示例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中的元素,判斷nums1中的元素是否在nums2中,如果在,將此元素加入到一個新的列表中,
    並在nums2中刪去這個元素。
  • 程式碼
class Solution:
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        seen = []
        for i in nums1:
            if i in nums2:
                seen.append(i)
                nums2.remove(i)   #只會移除其中碰到的一個元素

        return seen