[Python] [백준] 1764번 - 듣보잡 (set 사용)

듣보잡 

2 초

256 MB

60789

25400

19391

40.280%

문제

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

입력

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

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

출력

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

예제 입력 1 복사

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

예제 출력 1 복사

2
baesangwook
ohhenrie

---

---

1. 틀린코드 - 런타임 에러

import sys

n, m = map(int, sys.stdin.readline().rstrip().split())
arr_n = []
arr_m = []

for _ in range(n):
    arr_n.append(sys.stdin.readline().rstrip())

for _ in range(m):
    arr_m.append(sys.stdin.readline().rstrip())

result = sorted(list(arr_n & arr_m))

print(len(result))
for i in range(len(result)):
    print(result[i])

방법은 간단하다.

두 리스트로 입력받고

겹치는 원소들로 새로운 리스트(result)를 만들었다.

그리고 정렬해서 사전순으로 배열 되도록 하였다.

 

그러나 & 은 집합연산이라 set에서 사용할 수 있는 연산자다.

list 끼리의 연산에서 사용해서 틀렸다.

근데 왜 "틀렸습니다"가 아니라

"런타임 에러"가 나온 건지는 모르겠다...

---

2. 맞은 코드

import sys

n, m = map(int, sys.stdin.readline().rstrip().split())
arr_n = set()
arr_m = set()

for _ in range(n):
    arr_n.add(sys.stdin.readline().rstrip())

for _ in range(m):
    arr_m.add(sys.stdin.readline().rstrip())

result = sorted(list(arr_n & arr_m))

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

list가 아니라 set을 사용했다.

set을 사용했으므로 원소를 넣을 때 add()를 사용했다.

 

그리고 교집합 연산자를 사용하여 겹치는 것들로만

새로운 리스트를 만들고 정렬하였다. (끝)

 

* 교집합: & / 합집합: | / 차집합: - / 교집합 제외: ^

* set에 사용할 수 있는 메소드는 다음과 같다.