1. 程式人生 > >leetcode349:兩個陣列交集

leetcode349:兩個陣列交集

思想:

題目要求給定兩個陣列,編寫一個函式來計算它們的交集。

1.定義變數ans儲存兩個陣列的交集

2.將nums1和nums2去重

3.判斷nums1中的元素num是否在nums2中,若在則新增ans中,不在則繼續for迴圈

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        ans=[]
        nums1=set(nums1)
        nums2=set(nums2)
        for num in nums1:
            if num in nums2:
                ans.append(num)
        return ans