백준 7785 회사에 있는 사람 - 파이썬

 

 

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

7785번: 회사에 있는 사람

 

 

 

set 자료형을 이용해서 데이터를 다루는 문제이다.

데이터의 수가 최대 10 ** 6이어서 빠른 입출력을 사용하지 않거나 list를 사용하면 시간 초과가 난다.

 

at_company라는 set 자료형을 생성해 줬다.

한 줄씩 input을 받으면서 만약 출근을 했다면 at_company에 add 해주고, 퇴근을 했다면 at_company에서 remove 해준다.

최종적으로는 현재 회사에 남아있는 사람들만 at_company에도 남아 있을 것이다.

 

사전 배열 역순으로 출력하라고 했으니까 list로 옮겨 정렬을 해준다.

이름을 한 줄씩 출력해주면 된다.

 

 

from sys import stdin

N = int(input())
at_company = set() # 회사에 있는 사람 저장

for _ in range(N): # 입력 데이터 처리
    name, status = stdin.readline().rstrip().split()

    if status == "enter":
        at_company.add(name)
    elif status == "leave":
        at_company.remove(name)

# 답 정렬해서 출력
answer = list(at_company)
answer.sort(reverse=True)

for i in range(len(answer)):
    print(answer[i])