Coding Cantabile

[BOJ] #7785_회사에 있는 사람(Python3, 파이썬) 본문

Coding Test/BOJ

[BOJ] #7785_회사에 있는 사람(Python3, 파이썬)

Gracekim 2022. 12. 5. 13:59

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

티어

Silver V

문제

상근이는 세계적인 소프트웨어 회사 기글에서 일한다. 이 회사의 가장 큰 특징은 자유로운 출퇴근 시간이다. 
따라서, 직원들은 반드시 9시부터 6시까지 회사에 있지 않아도 된다.

각 직원은 자기가 원할 때 출근할 수 있고, 아무때나 퇴근할 수 있다.

상근이는 모든 사람의 출입카드 시스템의 로그를 가지고 있다. 이 로그는 어떤 사람이 회사에 들어왔는지, 나갔는지가 기록되어져 있다. 
로그가 주어졌을 때, 현재 회사에 있는 모든 사람을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 로그에 기록된 출입 기록의 수 n이 주어진다. (2 ≤ n ≤ 106) 다음 n개의 줄에는 출입 기록이 순서대로 주어지며, 
각 사람의 이름이 주어지고 "enter"나 "leave"가 주어진다. "enter"인 경우는 출근, "leave"인 경우는 퇴근이다.

회사에는 동명이인이 없으며, 대소문자가 다른 경우에는 다른 이름이다. 사람들의 이름은 알파벳 대소문자로 구성된 5글자 이하의 문자열이다.

출력

현재 회사에 있는 사람의 이름을 사전 순의 역순으로 한 줄에 한 명씩 출력한다.

예제 입력 및 출력 1

4
Baha enter
Askar enter
Baha leave
Artem enter
Askar
Artem

풀이

import sys
input = sys.stdin.readline

n = int(input())
com = dict()

for _ in range(n):
    name, con = map(str, input().split())

    if con == 'enter':
        com[name] = con
    else:
        del com[name]

for co in sorted(com, reverse=True):
    print(co)

처음에 리스트 자료형으로 풀었다가 시간 초과가 발생하였다. 기본적으로 리스트의 메소드들보다 딕셔너리의 대부분 메소드들의 시간복잡도가 더 작기때문에(O(1)) 딕셔너리로 접근하였다. 아마 마지막 for문에서 for __ in 리스트: 이 부분에서 딕셔너리는 존재 여부 확인을 할때의 시간복잡도가 O(1)이지만, 리스트는 O(n)이기 때문에 이 차이도 있었던 것 같다. 문제를 봤을 때 리스트와 딕셔너리를 둘 다 사용할 수 있을 것 같은 문제에는 웬만하면 딕셔너리를 사용하도록 해야겠다.