문제
김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.
듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.
출력
듣보잡의 수와 그 명단을 사전순으로 출력한다.
예제
나의 풀이1
import sys
input = sys.stdin.readline
N, M = map(int, input().split())
dic = {input().strip() : False for _ in range(N)}
dic = dict(sorted(dic.items()))
cnt = 0
for _ in range(M):
name = input().strip()
if name in dic:
dic[name] = True
cnt += 1
print(cnt)
for name in dic:
if dic[name]:
print(name)- 첫 번째 그룹의 명단을 딕셔너리 형태로 입력받고 사전순으로 정렬한다.
- 두 번째 그룹의 사람이 첫 번째 그룹에도 존재하는 경우 해당 쌍의 값을 `True`로 변환한다.
- 딕셔너리의 값이 `True`인 경우만 출력한다.
나의 풀이2
import sys
input = sys.stdin.readline
N, M = map(int, input().split())
a = set(input().strip() for _ in range(N))
b = set(input().strip() for _ in range(M))
result = sorted(a & b) # 교집합
print(len(result))
for name in result:
print(name)- 두 경우의 입력값을 각각 `set`에 추가해서 `&` 연산자로 교집합을 출력했다.
- `set`을 사용한 풀이 방법의 실행 속도가 조금 더 빨랐다.