[백준] 1764번 - 듣보잡

문제 설명

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

입력

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

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

출력

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

---

접근 방법

1. 기존에 작성한 코드

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

s1 = set()
for _ in range(N):
  s1.add(input())
s2 = set()
for _ in range(M):
  s2.add(input())

result = sorted(list(s1.intersection(s2)))
print(len(result))
print('\n'.join(result))

🔹 문제점

• set을 활용하여 교집합을 구하는 방식은 효율적이지만, 불필요한 set() 변환 과정이 있다.
• sorted(list())를 통해 리스트 변환 후 정렬하는데, 이는 sorted(set())로 더 간결하게 작성할 수 있다.

---

2. 개선된 코드

import sys

N, M = map(int, sys.stdin.readline().split())

db = set(sys.stdin.readline().strip() for _ in range(N))
bb = set(sys.stdin.readline().strip() for _ in range(M))

result = sorted(db & bb)

print(len(result))
print('\n'.join(result))

🔹 개선된 점

• sys.stdin.readline()을 사용하여 입력 속도를 향상시켰다.
• set을 바로 생성하면서 교집합을 구해 코드가 더 간결해졌다.
• sorted(list(s1 & s2)) 대신 sorted(s1 & s2)로 리스트 변환을 줄여 메모리 사용을 최적화했다.

---

✨ 핵심 정리

• set을 이용하면 중복 검사가 필요 없어지고, 교집합 연산(&)으로 쉽게 해결 가능하다.
• sys.stdin.readline()을 활용하면 입력 속도를 개선할 수 있다.
• 불필요한 리스트 변환을 줄이고, 가독성을 높이는 것이 효율적인 코드 작성의 핵심! 🚀