[알고리즘] 백준 1764 (파이썬 풀이)

문제

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

입력

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

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

출력

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

 

해결코드

import sys

n, m = map(int, sys.stdin.readline().split())
li = set()
li2 = set()
for i in range(n):
    li.add(sys.stdin.readline().rstrip())
for i in range(m):
    li2.add(sys.stdin.readline().rstrip())

result = sorted(list(li&li2))
print(len(result))
print(*result, sep="\n")

문제풀이

이 문제는 set()을 잘 이용하여 푸는 문제이다. 난 처음에 list를 사용했다가

시간초과가 나서 set()을 사용하였다.

문제 설명을 간단하게 하면 그냥 이름 리스트 두개를 받아 둘의 교집합을 찾고 출력하는 문제이다.

일단 n,m에 이름을 몇 개씩 받을 건지 입력받아준다.

그리고 li와 li2라는 set()을 만들어준다.(set을 써야 나중에 교집합을 이용할 수 있음)

그리고 각각 for문을 이용하여 n,m만큼 이름을 입력받아준다.

result에는 li와 li2의 교집합(&)을 리스트에 넣어주고 오름차순으로 정렬한 값을 넣어준다.

그리고 result의 길이를 출력해주고

다음줄에는 result를 unpack하여 줄바꿈 단위로 끊어 출력해준다.