leetcode之 Intersection of Two Arrays

来源:互联网 发布:万国数据公司怎么样 编辑:程序博客网 时间:2024/05/08 16:47

这题比较简单,是求2个不同list的公共部分。要求是元素必须是唯一的。先将2个list去重,然后从少的list开始循环求解。

class Solution(object):    def intersection(self, nums1, nums2):        """        :type nums1: List[int]        :type nums2: List[int]        :rtype: List[int]        """        if nums1 == [] or nums2 == []:            return []        result = []        nums1 = set(nums1)        nums2 = set(nums2)        if len(nums1) <= len(nums2):            for i in nums1:                if i in nums2:                    result.append(i)        else:            for i in nums2:                if i in nums1:                    result.append(i)        return result
0 0
原创粉丝点击