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

 

11651번: 좌표 정렬하기 2

 

문제

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

입력

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

출력

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

 

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

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

 

풀이

 

 

[백준 11650] 정렬 - 좌표 정렬하기

 

직전 문제의 코드에서 0과 1만 뒤집어 주면 문제를 해결할 수 있다.

 

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][1] == right[j][1]:
            if left[i][0] < right[j][0]:
            # 위 부분의 0과 1을 뒤집어주었다.
                L2.append(left[i])
                i+=1
            else:
                L2.append(right[j])
                j+=1
        elif left[i][1] < right[j][1]:
            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])