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

猜你喜欢

转载自blog.csdn.net/weixin_43160613/article/details/84102005