[Python] 백준 1764번 듣보잡 (실버5)

문제

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

입력

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

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

출력

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

예제 입력 1

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

예제 출력 1

2
baesangwook
ohhenrie

---

풀이

조건문을 잘 활용하는 문제입니다.

Point

• 조건문 (if)
• 반복문 (for)

---

소스코드

import sys
def input():
    return sys.stdin.readline().strip()

n, m = map(int, input().split())

s = set()
db = []

for _ in range(n):
    s.add(input())
for _ in range(m):
    name = input()
    if name in s:
        db.append(name)

db.sort()
print(len(db))
for n in db:
    print(n)

 

듣도 못한 사람을 저장해두고

그 중, 보도 못한 사람을 골라

알파벳 순으로 정리하고

차례로 출력하는 문제입니다.

 

Line 1~3

import sys
def input():
    return sys.stdin.readline().strip()

 

input()을 입력하는 라인 수가 많을 것으로 예상되면,

위와 같이 input() 함수를 sys.stdin.readline()으로 교체시켜주면, 실행시간이 빨라집니다.

'시간초과'의 원인이 된 적이 있기 때문에, 습관적으로 해두면 테스트 실행 한번을 아낄 수 있습니다

strip()은 입력값 뒤의 개행문자 '\n' (혹은 enter)를 제거하기 위함입니다.

Line 5~8

n, m = map(int, input().split())

s = set()
db = []

 

n, m 값을 입력값으로 받아주고, 듣도 못한 사람의 명단을 집합 s로 받아보겠습니다. (list도 관계 없음)

듣도 보도 못한 사람의 명단은 리스트 db에 저장해 보겠습니다. (나중에 sort를 하기 위해 list로 받습니다)

Line 10~15

for _ in range(n):
    s.add(input())
for _ in range(m):
    name = input()
    if name in s:
        db.append(name)

 

n명의 듣도 못한 분들을 n회 input() 마다 s에 추가시켜 줍니다.

 

m명의 보도 못한 분들을 m회 input()을 받습니다.

이 때, 듣도 못한 분이라면, 리스트 db에 추가합니다.

 

Line 17~20

db.sort()
print(len(db))
for n in db:
    print(n)

 

듣도 보도 못한 db를 알파벳 순으로 sort() 합니다.

 

db의 길이는 곧 듣도 보도 못한 분의 수입니다. 출력해 줍니다.

 

db에 있는 사람의 이름마다 출력을 해 줍니다.