[🥈5 / 백준 11650 / 파이썬] 좌표 정렬하기

 

11650번: 좌표 정렬하기

 

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

 

예제 입력 5 3 4 1 1 1 -1 2 2 3 3

예제 출력 1 -1 1 1 2 2 3 3 3 4

 

풀이

 

 

오름차순으로 정렬을 하되, x좌표가 같을 경우 y좌표를 오름차순으로 정렬하는 문제이다.

 

라이브러리의 정렬 함수를 이용하지 않고 직접 만들어본다고 했을 때,

필자는 이전에 사용했던 합병 정렬을 이용하면 될 것이라고 생각했다.

 

합병 정렬에서 합병을 할 때, x좌표가 서로 같은 지 비교하여 같다면

y좌표를 비교하여 합병해주는 과정만 추가해주면 된다.

 

import sys
def merge_sort(L=[]):
    if len(L) == 1:
        return L
    
    mid = len(L)//2
   
    left = merge_sort(L[:mid])
    right = merge_sort(L[mid:])
    
    i,j = 0,0
    L2 = []
    while ((i < len(left)) and (j < len(right))):
        if left[i][0] == right[j][0]:
            if left[i][1] < right[j][1]:
                L2.append(left[i])
                i+=1
            else:
                L2.append(right[j])
                j+=1
        # 이전 함수에서 위 과정이 새로 추가되었다.
        # x좌표가 같다면 y좌표를 비교해준다.
        elif left[i][0] < right[j][0]:
            L2.append(left[i])
            i+=1
        else:
            L2.append(right[j])
            j+=1
    while i < len(left):
        L2.append(left[i])
        i+=1
    while j < len(right):
        L2.append(right[j])
        j+=1
        
    return L2

a = [list(map(int,sys.stdin.readline().split())) for i in range(int(sys.stdin.readline()))]
a = merge_sort(a)
for i in a:
    print(i[0],i[1])