leetcode之 Intersection of Two Arrays

最新推荐文章于 2020-12-20 13:47:46 发布

原创最新推荐文章于 2020-12-20 13:47:46 发布 · 454 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #去重

本文介绍了一种简单有效的算法来找出两个列表的交集部分，确保结果中元素唯一且不重复。通过使用集合数据结构去除重复项并进行比较，提高了算法效率。

这题比较简单，是求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