leetcode之 Intersection of Two Arrays

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