백준 15970 화살표 그리기 (파이썬)

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

15970번: 화살표 그리기

 

처음엔 가는 방향과 오는 방향을 따로 반복문을 돌려 문제를 풀어봤지만,

문제의 핵심은 방향 구분 없이 동일한 색의 점 중에서 가장 최소거리인 점을 찾기만 하면 되는 문제입니다.

정렬을 시행한 후 반목문과 최솟값을 비교하는 min 함수를 이용하여 각 점의 최소거리를 구하면 되는 문제입니다.

 

 

N = int(input())
lst = []
for i in range(N):
    a, b = map(int, input().split())
    lst.append((a, b))
 
val = 0
lst.sort(key=lambda x: x[0])
 
for i in range(N):
    sub_lst = []
    for j in range(N):
        if i != j and lst[i][1] == lst[j][1]:
            sub_lst.append(abs(lst[i][0]-lst[j][0]))
    val += min(sub_lst)
print(val)