백준 11651번 (python) - 좌표 정렬하기2

좌표 정렬하기 2

백준 > 단계별로 풀어보기 > 정렬

문제

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(sys.stdin.readline())
array = []

for i in range(N):
    array.append(list(map(int, sys.stdin.readline().split())))

array = sorted(array,key=lambda x : (x[1],x[0])) # key = lamda

for i in array :
    print(i[0],i[1])

sorted 함수에 key로 lamda를 인자로 전달가능하다.

lamda함수는 JS에서 사용했던 익명함수랑도 비슷한개념이다.