백준 7785 <회사에 있는 사람> Python

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

7785번: 회사에 있는 사람

---

입력이 긴 문제의 경우 Input() 보다 sys.stdin.readline() 을 사용하는 것이 좋습니다. 그리고 readline 뒤에 \n 같은 공백 문자도 같이 입력되니 strip() 을 사용하면 깔끔하게 받아올 수 있습니다. 문제는 Dict 타입을 사용하면 쉽게 풀 수 있습니다.

---

import sys

from collections import defaultdict

def solution(n, data_list):

    # 출입 명부 딕셔너리

    in_out_dict = defaultdict(int)

    # 순회하면서 상태 최신화

    for name, state in data_list:

        if state == 'enter':

            in_out_dict[name] = 1

        else:

            in_out_dict[name] = 0

    # 딕셔너리 정렬

    in_out_dict = dict(sorted(in_out_dict.items(), reverse=True))

    # 순회하며 출력

    for key, value in in_out_dict.items():

        if value:

            print(key)    

n = int(input())

data_list = [sys.stdin.readline().strip().split(' ') for _ in range(n)]

solution(n, data_list)