(Python) 백준 11650번 - 좌표 정렬하기

문제 링크 : https://www.acmicpc.net/problem/11650

11650번: 좌표 정렬하기

풀이

sort() 함수에 key로 lambda 함수를 넘겨주게 되면, 이 함수의 반환값을 기준으로 순서대로 정렬하게 된다.

여기서는 x[0], x[1] 순으로, 즉 리스트의 첫번째 요소인 x좌표로 정렬 후에 두번째 요소인 y좌표를 기준으로 정렬하게 되는 것이다. 

import sys
num = int(sys.stdin.readline())
xys = []
for _ in range(num):
    xy = list(map(int, sys.stdin.readline().split()))
    xys.append(xy)
for x, y in sorted(xys, key=lambda x: (x[0], x[1])): # 리스트의 첫번째 element인 x좌표를 기준으로 정렬 후 두번째 element인 y좌표를 기준으로 정렬
    print(x, y)