Python 백준 1764 - 듣보잡

백준 링크

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

1764번: 듣보잡

---

문제

입력

출력

입출력 예시

---

제출한 코드

1차 제출 코드

import sys

N, M = map(int, sys.stdin.readline().split())
data = []
answer = []

for i in range(N+M):
    check = sys.stdin.readline().strip()
    if check in data:
        answer.append(check)
    else:
        data.append(check)

answer=list(set(answer))
print(len(answer))
print(*sorted(answer), sep = "\n")

Pycharm IDE 환경에서는 기능을 수행하는데 문제가 없었다.

 

1. N과 M의 값 입력받기

2. N+M번만큼 data에 값 넣기

3. data에 값을 넣기 전 이미 존재하는 데이터인지 확인하기

4. 이미 존재하는 경우일 때 answer에 값을 저장하기

5. 저장된 answer의 중복을 삭제하고 정렬해서 출력하기

 

내가 생각한 로직은 위와 같고 시간 초과가 떴다.

 

2차 제출 코드

 

import sys

N, M = map(int, sys.stdin.readline().split())
data = set()
answer_set = set()

for i in range(N + M):
    check = sys.stdin.readline().strip()
    if check in data:
        answer_set.add(check)
    else:
        data.add(check)

answer = sorted(answer_set)

print(len(answer))
print(*answer, sep="\n")

 

시간을 더 줄이기 위해서 for문으로 값을 입력받는 건 수정이 힘들 거 같아 데이터를 저장하는 부분에서 list대신 set을 이용해서 시간을 줄이고 문제를 통과했다.