[Python] 백준 14402번

14402. 야근 (silver 4)

# 자료구조

영선회사는 야근이 매우 잦은 회사이다. 다행히도 야근수당을 챙겨 주기는 하는데, 사장 nein은 매우 악덕한 사장이기 때문에 최대한 야근 수당을 안 챙겨주려 한다.

영선회사이 야근인지 확인하는 방법은 다음과 같다. 매우 이른 아침부터 퇴근시간까지 출입기록이 주어진다. 만약, 들어간 기록은 없는데, 나온 기록이 있다면 회사에서 야근을 한 것으로 기록된다. 또한 회사에 들어갔는데, 퇴근시간까지 나오지 않는다면, 이 또한 야근을 하는 것으로 기록된다. 출입기록에 따라서 두 기준 다 만족할 수도 있는데, 이는 야근을 두 번 한 것으로 인정된다. 하지만, 회사에 계속 있었지만 나오거나 들어가지를 않아 출입기록이 없다면 야근을 한 것으로 인정하지 않는다.

그리고 이 출입기록에는 치명적인 단점이 있는데, 동명이인을 구별하지 못한다는 것이다. 이름이 같은 경우, 같은 사람으로 인식하며, 둘다 회사에 남아있거나 들어간 기록 없이 나온 기록이 있는 식으로, 명확히 보이는 경우에만 복수로 야근이 인정된다.

영선회사의 오늘의 출입기록이 주어진다. 이 기록을 바탕으로 모든 직원의 야근 횟수의 합을 출력하시오.

 

입력

첫째 줄에는 출입기록의 개수 q가 주어진다.( 1≤q≤100,000)

다음 q줄에는 이름 s와 들어간지 나간지에 대한 정보 p가 주어지는데, p는 ‘+’일 때 들어간 것이고, ‘-‘일 때 나간 것이다. 이름의 길이는 30자를 넘지 않는다. 이름은 소문자로만 이루어져 있다.

 

출력

출입기록을 바탕으로 모든 직원의 야근 횟수의 합을 출력하시오.

 

 

예제 입력1

7
nein +
nein -
nein -
nein -
nein +
nein +
nein +

예제 출력1

5

 

 

 

풀이)

우선 출입 기록의 개수를 q 변수에 저장해주었다. 그 다음 딕셔너리와 overtime 변수를 선언해주었다.

for문을 사용해서 출입 기록의 개수만큼 반복할 때, record에 출입 기록을 리스트 형태로 저장해주었다.

입력한 기록 중 0번째 인덱스를 name에 저장하고, 1번째 인덱스를 data에 저장해주어 if elif문을 써주었다.

만약 data의 값이 -인 경우에는 회사에서 나온 경우이기 때문에 두 가지 경우로 또 나눠줄 수 있다.

1. check 딕셔너리에 이름이 없는 경우 -> 야근한 경우
2. check 딕셔너리에 이름이 있는 경우 -> 야근을 하지 않고 퇴근한 경우

따라서 이에 맞게 1번의 경우에는 야근한 사람의 수를 세는 변수인 overtime에 1을 더해주었고, 2번의 경우에는 name 인덱스에 해당하는 value값을 1 빼주었다.

elif문을 사용해서 data가 +인 경우에는 회사에 들어간 경우이다. name이 딕셔너리에 없는 경우 1로 value값을 저장해주었고, name이 이미 딕셔너리에 존재하는 경우에는 1을 더해주었다. 이때 name 인덱스에 대해 value의 값이 0이 아닌 경우도 overtime과 함께 야근한 케이스에 속하기 때문에 마지막 야근하는 직원의 수를 출력할 때에는 check 딕셔너리의 value값과 overtime을 더해준 값을 출력해주었다.

 

import sys
input = sys.stdin.readline

q = int(input())  # 출입 기록의 개수

check = {} 
overtime =0    
for i in range(q):
  enter = list(input().split())  # 기록 
  name, data = enter[0], enter[1]
  if data == '-':     # 회사에서 나옴
    if name not in check or check[name] == 0:  # 야근한 경우
      overtime +=1
    elif name in check:    # 야근 안 한 경우
      check[name] -=1
  elif data == '+':    # 들어간 경우
    if name not in check:    
      check[name] = 1    # 들어갔다고 등록
    else:
      check[name] += 1   

# 야근하는 직원 수
print(sum(check.values()) + overtime)