Day21:给定两个数组,编写一个函数来计算它们的交集。
eg1 : input : nums1 = [1,2,2,1], nums2 = [2,2]
output : [2,2]
eg2 : input : nums1 = [4,9,5], nums2 = [9,4,9,8,4]
output : [4,9]
#如果要单纯的求出两个列表里面相同的元素,直接用set()即可
#参考力扣大佬写的代码,实在是太妙了!!!
方法一:
1 def intersect(nums1, nums2): 2 3 #求出两个列表共同的元素(唯一) 4 #以【1,2,2,1】和【2,2】为例,inter = {2} 5 inter = set(nums1) & set(nums2) 6 l = [] 7 for i in inter: 8 #i的个数为2 9 l += [i] * min(nums1.count(i), nums2.count(i)) 10 11 return l 12 13 if __name__ == "__main__": 14 15 nums1 = [1,2,2,1] 16 nums2 = [2,2] 17 print(intersect(nums1,nums2)) 18
方法二:双指针,先排序后进行遍历。
1 def intersect(nums1, nums2): 2 nums1.sort() 3 nums2.sort() 4 r = [] 5 left, right = 0, 0 6 while left < len(nums1) and right < len(nums2): 7 if nums1[left] < nums2[right]: 8 left += 1 9 elif nums1[left] == nums2[right]: 10 r.append(nums1[left]) 11 left += 1 12 right += 1 13 else: 14 right += 1 15 return r
输出结果:[2,2]
#补充关于Python里面的交集,并集和差集运算
#Python交集
1 print([val for val in nums1 if val in nums2]) 2 print(list(set(nums1) & set(nums2))) 3 print(list(set(nums1).intersection(set(nums2))))
#Python并集
1 print(list(set(nums1).union(set(nums2)))) 2 print(list(set(nums1) or set(nums2)))
#Python差集
1 print(list(set(nums1).difference(set(nums2)))) 2 #差集还可以是并集减去交集 3 set_union = list(set(nums1).union(set(nums2))) 4 set_inter = list(set(nums1).intersection(set(nums2))) 5 set_diff = [i for i in set_union if i not in set_inter] 6 print(set_diff)