티스토리 뷰

728x90

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

 

15975번: 화살표 그리기

직선위에 $N$개의 점들이 주어지고 각 점은 $N$개의 색깔 중 하나를 가진다. 편의상, 색깔은 1부터 $N$까지의 수로 표시 하고, 점들의 좌표는 모두 다르다. 각 점 $p$에 대해서, $p$에서 시작하는 직선

www.acmicpc.net

 

같은 색상끼리 최소거리 합을 구하는 문제입니다.

정렬했을 시 최소거리는 기준점으로부터 바로 앞, 뒤 값과의 차가 최솟값이기 때문에

색상인 2열 입력값을 기준으로 정렬을 하여 구현합니다.

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
n = int(input())
lst = [[] for _ in range(n)]
for i in range(n):
    a, b = map(int, input().split())
    lst[i] = [a, b]
lst.sort(key=lambda x: (x[1], x[0]))
res = 0
for i in range(n):
    s = 2 * 10**9 + 1
    if 0 <= i < n - 1 and lst[i][1] == lst[i + 1][1]:
        s = min(s, abs(lst[i][0] - lst[i + 1][0]))
    if 0 < i < n and lst[i][1] == lst[i - 1][1]:
        s = min(s, abs(lst[i][0] - lst[i - 1][0]))
    if s == 2 * 10**9 + 1:
        s = 0
    res += s
print(res)
cs

 

 

댓글