[백준] 1764 : 듣보잡 (Python/파이썬)

문제 링크

https://www.acmicpc.net/problem/1764

문제

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

입력

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

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

출력

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

예제

입력

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

출력

2
baesangwook
ohhenrie

---

정답 및 풀이

import sys
input = sys.stdin.readline

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

nlist = set(input().strip() for _ in range(n))
mlist = set(input().strip() for _ in range(m))

result = sorted(list(nlist & mlist))

print(len(result))
for name in result:
    print(name)

1. import sys / input = sys.stdin.readline
   • 많은 양의 입력을 빠르게 받기 위해 sys.stdin.readline() 사용한다.
2. n, m = map(int, input().split())
   • 듣도 못한 사람 수(n), 보도 못한 사람 수(m)를 입력받는다.
3. nlist = set(input().strip() for _ in range(n))
   • n명의 듣도 못한 사람 이름을 입력받아 집합(set) 으로 저장한다.
   • strip()은 개행문자 제거한다.
   • 집합을 사용하면 중복 제거 + 탐색 속도 O(1).
4. mlist = set(input().strip() for _ in range(m))
   • 보도 못한 사람 명단을 집합으로 저장한다.
5. result = sorted(list(nlist & mlist))
   • &는 교집합 연산자.
     • 두 집합에 공통으로 존재하는 이름만 추출한다.
   • list()로 변환 후 sorted()로 사전순 정렬한다.
6. print(len(result))
   • 듣보잡의 인원 수 출력한다.
7. for name in result: print(name)
   • 듣보잡 명단을 한 줄씩 출력한다.

다른 풀이

import sys
input = sys.stdin.readline

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

name_count = {}

for _ in range(n):
    name = input().strip()
    name_count[name] = 1  # 듣도 못한 사람 기록

result = []

for _ in range(m):
    name = input().strip()
    if name in name_count:  # 이미 듣도 못한 명단에 있으면
        result.append(name)

result.sort()

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

---

새롭게 배운 내용 및 느낀점

• 집합(set)
  • 중복이 없는 데이터 구조.
  • in 연산 시 시간 복잡도 O(1).
  • 교집합(&), 합집합(|), 차집합() 연산 지원.
• 교집합 연산자 &
  • 두 집합의 공통 요소를 추출.
  • 예: {1,2,3} & {2,3,4} → {2,3}