백준 - 듣보잡(1764)

String + Binary Search

문제

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

입력

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

듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.

출력

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


import sys

M, N = map(int, sys.stdin.readline().split())
a = []
answer = []

for i in range(M):
    a.append(input().strip())
    
for i in range(N):
    b = input().strip()
    if b in a:
        answer.append(b)
        
print(len(answer))
for letter in answer:
    print(letter)
    
# 시간초과

아래는 set을 이용한 풀이

import sys

M, N = map(int, sys.stdin.readline().split())
a = set([])
b = set([])

for i in range(N):
    a.add(sys.stdin.readline().strip())
    
for i in range(M):
    b.add(sys.stdin.readline().strip())
    
mutual = list(a & b) # 아래에서 원소들을 출력해야 한다. 따라서 a & b도 set이니까 배열로 1번 더 감싼다.
print(len(mutual))
for m in sorted(mutual): # 사전순 출력 => sorted() 사용
    print(m)
    
# a = set([]) => set은 주로 배열을 감싸서 사용한다.
# 추가: a.add(), 삭제: a.remove()
# 중복 없이 요소를 추가하기에 순서는 없다

좋은 웹페이지 즐겨찾기