백준 1269번: 대칭 차집합. Python 파이썬 답안, 백준 코딩테스트 파이썬 풀이

백준 1269번: 대칭 차집합. Python 파이썬 답안, 백준 코딩테스트 파이썬 풀이

 

 

문제

자연수를 원소로 갖는 공집합이 아닌 두 집합 A와 B가 있다. 이때, 두 집합의 대칭 차집합의 원소의 개수를 출력하는 프로그램을 작성하시오. 두 집합 A와 B가 있을 때, (A-B)와 (B-A)의 합집합을 A와 B의 대칭 차집합이라고 한다.

예를 들어, A = { 1, 2, 4 } 이고, B = { 2, 3, 4, 5, 6 } 라고 할 때,  A-B = { 1 } 이고, B-A = { 3, 5, 6 } 이므로, 대칭 차집합의 원소의 개수는 1 + 3 = 4개이다.

입력

첫째 줄에 집합 A의 원소의 개수와 집합 B의 원소의 개수가 빈 칸을 사이에 두고 주어진다. 둘째 줄에는 집합 A의 모든 원소가, 셋째 줄에는 집합 B의 모든 원소가 빈 칸을 사이에 두고 각각 주어진다. 각 집합의 원소의 개수는 200,000을 넘지 않으며, 모든 원소의 값은 100,000,000을 넘지 않는다.

출력

첫째 줄에 대칭 차집합의 원소의 개수를 출력한다.

답안

def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    
    # 첫 줄: 집합 A와 B의 원소 개수
    nA = int(data[0])
    nB = int(data[1])
    
    # 둘째 줄: 집합 A의 원소
    A = set(map(int, data[2:2+nA]))
    
    # 셋째 줄: 집합 B의 원소
    B = set(map(int, data[2+nA:2+nA+nB]))
    
    # 대칭 차집합의 원소 개수 계산
    symmetric_difference = A.symmetric_difference(B)
    print(len(symmetric_difference))

if __name__ == "__main__":
    main()