[백준][Python][15970][구현] 화살표 그리기

문제 : https://www.acmicpc.net/problem/15970

15970번: 화살표 그리기

구현 문제다.

 

1. 우선 입력을 정렬한다. 겹치는게 없으므로 정렬 조건은 설정 안해도 된다.
2. 입력받은 좌표를 하나하나 돌아가면서
   1. 현재 좌표보다 왼쪽에 있는 것중에 색이 같은것을 찾는다.
   2. 현재 좌표보다 오른쪽에 있는 것중에 색이 같은것을 찾는다.
3. tmpret에 왼쪽, 오른쪽 값이 나오는데 그 중에서 작은것을 골라 답에 더해준다

 

import sys

n = int(sys.stdin.readline())

d = []
for _ in range(n):
    d.append([int(i) for i in sys.stdin.readline().split()])


d.sort()
ret = 0

for i in range(n):
    tmploc, tmpclr = d[i]
    tmp = i-1
    tmpret = [9912341234, 9912341234]
    # left
    while tmp >= 0:
        if d[tmp][1] == tmpclr:
            tmpret[0] = abs(tmploc - d[tmp][0])
            break
        else:
            tmp -= 1

    # right

    tmp = i+1
    while tmp < n:
        if d[tmp][1] == tmpclr:
            tmpret[1] = abs(tmploc - d[tmp][0])
            break
        else:
            tmp += 1
    ret += min(tmpret)


print(ret)