[백준 2852] NBA 농구(파이썬 풀이)

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

2852번: NBA 농구

문제

동혁이는 NBA 농구 경기를 즐겨 본다. 동혁이는 골이 들어갈 때 마다 골이 들어간 시간과 팀을 적는 이상한 취미를 가지고 있다.

농구 경기는 정확히 48분동안 진행된다. 각 팀이 몇 분동안 이기고 있었는지 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 골이 들어간 횟수 N(1<=N<=100)이 주어진다. 둘째 줄부터 N개의 줄에 득점 정보가 주어진다. 득점 정보는 득점한 팀의 번호와 득점한 시간으로 이루어져 있다. 팀 번호는 1 또는 2이다. 득점한 시간은 MM:SS(분:초) 형식이며, 분과 초가 한자리 일 경우 첫째자리가 0이다. 분은 0보다 크거나 같고, 47보다 작거나 같으며, 초는 0보다 크거나 같고, 59보다 작거나 같다. 득점 시간이 겹치는 경우는 없다.

출력

첫째 줄에 1번 팀이 이기고 있던 시간, 둘째 줄에 2번 팀이 이기고 있던 시간을 출력한다. 시간은 입력과 같은 형식(MM:SS)으로 출력한다.

예제 입력 1 

1
1 20:00

예제 출력 1 

28:00
00:00

예제 입력 2 

3
1 01:10
2 21:10
2 31:30

예제 출력 2 

20:00
16:30

예제 입력 3 

5
1 01:10
1 02:20
2 45:30
2 46:40
2 47:50

예제 출력 3 

45:30
00:10

제출 코드

n = int(input())
flag, a, b = 0,0,0
end = 48*60
for _ in range(n):
    i, time = input().split()
    min, sec = map(int, time.split(':'))
    time = min*60 + sec
    if i=='1':
        # a가 이기기 시작
        if flag == 0:
            a += end - time
        flag += 1
        # b가 이기고있다 비김
        if flag == 0:
            b -= end - time
    else:
        # b가 이기기 시작
        if flag == 0:
            b += end - time
        flag -= 1
        # a가 이기고있다가 비김
        if flag == 0:
            a -= end - time

print('{:0>2}:{:0>2}'.format(a//60, a%60))
print('{:0>2}:{:0>2}'.format(b//60, b%60))

첫번째 팀 a, 두번째 팀 b

전체 시간 48분에서 이기고있는 시간은 더하고, 지고있는 시간은 빼는 방식으로 구했다.