门| GATE-CS-2014-(Set-2) |第 35 题

这道题目给出了两个集合A和B，并要求我们求出操作A和B之后得到的集合的大小。其中操作分为3种：

• 并集操作：将A和B的所有元素放入一个新的集合中，并且不重复。
• 交集操作：将A和B中共有的元素放入一个新的集合中。
• 差集操作：将只属于A而不属于B的元素放入一个新的集合中。

题目中给出了集合A和集合B的元素个数，并将它们作为输入。

解题方法

因为A和B要进行的操作都很简单，所以我们可以直接遍历两个集合来得到操作的结果。

并集操作

对于并集操作，我们可以将集合A和集合B合并成一个新的集合，而不重复地添加元素。

union_set = set(A)
for elem in B:
    union_set.add(elem)
    
union_size = len(union_set)

交集操作

对于交集操作，我们可以循环遍历集合A中的每一个元素，查看它是否在集合B中出现。如果是，则将这个元素添加到新的集合中。

intersection_set = set()
for elem in A:
    if elem in B:
        intersection_set.add(elem)

intersection_size = len(intersection_set)

差集操作

对于差集操作，我们可以循环遍历集合A中的每一个元素，查看它是否在集合B中出现。如果不是，则将这个元素添加到新的集合中。

difference_set = set()
for elem in A:
    if elem not in B:
        difference_set.add(elem)

difference_size = len(difference_set)

最后，我们将三种操作得到的元素个数返回即可。

return union_size, intersection_size, difference_size

整合代码

def find_set_sizes(A, B):
    union_set = set(A)
    for elem in B:
        union_set.add(elem)
    
    intersection_set = set()
    for elem in A:
        if elem in B:
            intersection_set.add(elem)
    
    difference_set = set()
    for elem in A:
        if elem not in B:
            difference_set.add(elem)
    
    return len(union_set), len(intersection_set), len(difference_set)

该函数会返回三个整数，分别表示A和B的并集、交集和差集的元素个数。