듣보잡

07 Jul 2022 in Algorithm on Set, 교집합

백준 Class3 듣보잡 , Silver4

solved_ac[Class3] 듣보잡

문제

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

입력

첫째 줄에 듣도 못한 사람의 수 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트

처음 문제에 접근하였을 때 단순하게 “문자열 비교”와 “정렬”만 사용하여 풀었다.

결과는 TLE

2트 (정답) , 3948ms

1차원 배열을 다루는 데 시간초과가 발생하기 때문에 Hash map을 이용하여 O(1)의 시간복잡도로 접근해야겠다고 생각했다.

다른 사람 아이디어, 4012ms

와 … 교집합을 이용하면 더욱 간단히 풀 수 있었다.

시간을 측정 해보았는데 크게 차이 나지는 않았다.

다른 사람들의 풀이를 참조해본 결과 이 문제에서 요구하는 정석적인 풀이는 다음과 같다.

⭐️ [두 개의 문자열들 중에서 공통 문자열을 추릴 때 set의 교집합을 사용할 수 있다.] ⭐️

---

CODE

1트

 hear, see = map(int,input().split())

lst_hear = []
for _ in range(hear):
    lst_hear.append(input())

lst_answer = []
for _ in range(see):
    name = input()
    if name in lst_hear:
        lst_answer.append(name)

lst_answer.sort()
print(len(lst_answer))
for answer in lst_answer:
    print(answer)

2트 (정답) , 3948ms

from collections import defaultdict

hear, see = map(int,input().split())

name_count = defaultdict(int)

for _ in range(hear+see):
    name = input()
    name_count[name] += 1

lst_answer = []
for name in name_count:
    if name_count[name] >= 2:
        lst_answer.append(name)

lst_answer.sort()

print(len(lst_answer))
for answer in lst_answer:
    print(answer)

다른 사람 아이디어, 4012ms

hear, see = map(int,input().split())

a = set()
for i in range(hear):
    a.add(input())

b = set()
for i in range(see):
    b.add(input())

result = sorted(list(a&b))

print(len(result))

for i in result:
    print(i)