[백준] 11651 좌표 정렬하기2 python

문제

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

입력

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

출력

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

코드

import sys
n = int(input())
arr = []
for i in range(n):
    x,y = map(int,sys.stdin.readline().rstrip().split())
    arr.append([x,y])
arr.sort(key = lambda a: (a[1],a[0]) )
for x,y in arr:
    print(x,y)

풀이

sort()에서 key 값만 잘 활용하면 쉽게 풀 수 있다.

key값을 쌍으로 줄 때, (첫 번째 조건, 두 번째 조건) 기준으로 주면 된다.