[리스트/딕셔너리] 백준 - 7785번: 회사에 있는 사람 (파이썬)

Intro

처음에 리스트로 구현하는 방향으로 접근했다가 계속 틀렸다고 떠서 결국 딕셔너리로 구현해서 해결한 문제이다.

(매니저님: 리스트로 하실 줄 알았는데 의외네요)

 

그래서 이번에는 내가 구현한 코드를 정리하고, 리스트로 다시 구현해 보겠다.

---

문제

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

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

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

 

입력

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

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

 

출력

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

 

예제 입력

4
Baha enter
Askar enter
Baha leave
Artem enter

예제 출력

Askar
Artem

 

 

내 풀이

n = int(input())
l_dic = {} # 사용자: 로그

for i in range(n):
    name, log = map(str, input().split()) # 사용자, 로그 입력
    # 사용자 value 에 log 기록, enter 이후에 leave 로그가 찍히면 값이 leave로 덮어 씌워지는 걸 이용
    l_dic[name] = log

    # value 값이 leave 가 된 사용자 딕셔너리에서 제거
    if l_dic[name] == 'leave':
        del l_dic[name]

#이름을 기준으로 내림차순 정렬
key = sorted(l_dic.keys(), reverse=True)

for i in key:
    print(i)

 

받은 피드백: key 값들을 모두 꺼내서 정렬하는 부분 최적화 필요

 

다시 리스트를 사용해서 구현해보자.

n = int(input())
l_set = set()

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

    if log == 'enter':
        l_set.add(name)
    else:
        l_set.remove(name)

list = list(l_set)
list.sort(reverse=True)

for i in list:
    print(i)

전체적인 흐름은 비슷하지만 변경된 부분이 있다.

 

- 딕셔너리에 name과 log를 모두 세팅 -> value 값이 leave인 key 삭제

  -> log가 enter인 name을 set에 모두 세팅 -> enter가 아닌 name 삭제

 

기껏 수정했더니 더 느리다.... 😥 왜지