백준 알고리즘 1764. 듣보잡

2024. 3. 19. 16:16코딩 테스트/백준

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

 

1764번: 듣보잡

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다.

www.acmicpc.net

 

전체 코드

import sys


def solution(not_seen, not_heard):
    return sorted(list(set(not_seen) & set(not_heard)))


not_seen = []
not_heard = []
N, M = map(int, input().split())
for _ in range(N):
    not_seen.append(sys.stdin.readline().strip('\n'))

for _ in range(M):
    not_heard.append(sys.stdin.readline().strip('\n'))

results = solution(not_seen, not_heard)
print(len(results))
for r in results:
    print(r)

 

두개의 리스트를 set으로 변경하여 & 연산을 하면 교집합을 간단히 얻을 수 있다.