시간제한 : 2초
메모리 제한 : 256MB
김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.
첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.
듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.
듣보잡의 수와 그 명단을 사전순으로 출력한다.
3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton
2
baesangwook
ohhenrie
import sys
N, M = map(int, sys.stdin.readline().strip().split())
set1 = set()
set2 = set()
# 듣도 못한 사람
for _ in range(N):
set1.add(sys.stdin.readline().strip())
# 보도 못한 사람
for _ in range(M):
set2.add(sys.stdin.readline().strip())
# 듣보잡들
result = sorted(list(set1 & set2))
# 듣보잡 수
print(len(result))
# 듣보잡 이름
for x in result:
print(x)
시간 : 112ms
메모리 : 44060KB
이 문제의 경우 교집합을 구하는 문제이므로 집합 set()
을 이용해 문제를 풀었다.
먼저 듣도 못한 사람과 보도 못한 사람을 set1
, set2
에 저장했다.
그 후 교집합 연산자인 &
를 이용해 듣도 못한 사람(set1)과 보도 못한 사람(set2)의 교집합을 사전순으로 정렬해 result
에 저장했다.
그리고 저장된 result
를 이용해 듣보잡의 수와 듣보잡의 이름들을 출력해줬다.