[파이썬] 백준 - 좌표 정렬하기 - 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

작성한 코드

import sys
from functools import cmp_to_key

N = int(sys.stdin.readline())
rtn = []
for _ in range(N):
    rtn.append(list(map(int, sys.stdin.readline().strip().split(' '))))

def compare(a, b):
    if a[0] < b[0]:
        return -1
    elif a[0] == b[0] and a[1] < b[1]:
        return -1
    else:
        return 0
for i in sorted(rtn, key=cmp_to_key(compare)):
    print(' '.join(map(str, i)))

백준에서 확인

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

11650번: 좌표 정렬하기