Python活力练习Day21

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()即可

#参考力扣大佬写的代码，实在是太妙了！！！

 方法一：

def intersect(nums1, nums2):
    
    #求出两个列表共同的元素（唯一）
    #以【1,2,2,1】和【2,2】为例，inter = {2}
    inter = set(nums1) & set(nums2)  
    l = []
    for i in inter:
        #i的个数为2
        l += [i] * min(nums1.count(i), nums2.count(i))

    return l

if __name__ == "__main__":

    nums1 = [1,2,2,1]
    nums2 = [2,2]
    print(intersect(nums1,nums2))
    

 方法二：双指针，先排序后进行遍历。

def intersect(nums1, nums2):
    nums1.sort()
    nums2.sort()
    r = []
    left, right = 0, 0
    while left < len(nums1) and right < len(nums2):
        if nums1[left] < nums2[right]:
            left += 1
        elif nums1[left] == nums2[right]:
            r.append(nums1[left])
            left += 1
            right += 1
        else:
            right += 1
    return r

 

输出结果：[2,2]

 

#补充关于Python里面的交集，并集和差集运算

#Python交集

 

print([val for val in nums1 if val in nums2])
print(list(set(nums1) & set(nums2)))
print(list(set(nums1).intersection(set(nums2))))

 

#Python并集

print(list(set(nums1).union(set(nums2))))
print(list(set(nums1) or set(nums2)))

#Python差集

print(list(set(nums1).difference(set(nums2))))
#差集还可以是并集减去交集
set_union = list(set(nums1).union(set(nums2)))
set_inter = list(set(nums1).intersection(set(nums2)))
set_diff = [i for i in set_union if i not in set_inter]
print(set_diff)