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