[백준 1764] 듣보잡

문제
듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.

입력
첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 영어 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.

출력
듣보잡의 수와 그 명단을 사전순으로 출력한다.

---

접근

두 개의 그룹에 모두 속한 이름을 출력하는 것이다. 집합으로 표현했을 때 교집합인 부분을 출력해야하며 파이썬에서 set()을 통해 집합을 표현한다.

list 형의 그룹들을 set 형으로 변경하여 & 연산을 사용하면 두 그룹에 포함된 이름이 나타난다.

 

구현

공백없이 이름을 입력받아 각각 A와 B에 저장한다.

N, M = map(int, input().split())

A = [input().strip() for _ in range(N)]
B = [input().strip() for _ in range(M)]

 

A와 B를 set형으로 변형하여 교집합을 취한다.

그렇게 되면 result에 두 그룹에 포함된 이름을 리스트형태로 저장된다.

result = list(set(A) & set(B))

 

이후 오름차순으로 정렬 후에 출력하도록 한다.

result.sort()
print(len(result))
for name in result:
    print(name)

---

전체 코드

import sys
input = sys.stdin.readline

N, M = map(int, input().split())

A = [input().strip() for _ in range(N)]
B = [input().strip() for _ in range(M)]

result = list(set(A) & set(B))
result.sort()
print(len(result))
for name in result:
    print(name)