백준

[python]은?행 털!자 1 - 26267

청원뿔세포 2022. 12. 19. 20:43

두번째 줄부터 입력받을 때 배열에 값들을 배열로 묶어 2중배열로 담아주었다.

이 때 시간 T - 좌표 X 를 한 값을 추가적으로 담아주었다.

시우는 좌표가 증가는 방향으로만 이동해야하기 때문에 멈출 수 없다.

T-X를 해준 값이 똑같은 은행만 방문할 수 있다.

T-X가 같은 값을 가진 은행들을 각각 합하여 가치가 높은 것을 출력해주었다.

import sys
n = int(input())
arr = []
for i in range(n):
    arr.append(list(map(int,sys.stdin.readline().split())))
    arr[i].append(arr[i][1]-arr[i][0])
timeset = {}
for  i in range(n):
    if arr[i][3] in timeset.keys():
        timeset[arr[i][3]].append(arr[i][2])
    else:
        timeset[arr[i][3]]=[arr[i][2]]
re = 0
for i in timeset:
    if re < sum(timeset[i]):
        re = sum(timeset[i])
print(re)