View

[Python] Baekjoon 백준 11651번 좌표 정렬하기 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(input())

num = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]

num.sort(key=lambda x: (x[1], x[0]))

[print(i[0], i[1]) for i in num]

 

num.sort(key=lambda x: (x[1], x[0]))

sort() 함수로 사용하여

 

key=파라미터 를 넘겨주게되고

 

파라미터에 lambda 를 이용한다

 

여기에 두번째 요소( x[1] ) 먼저, 그다음 첫번째 요소( x[0] ) 순서로 오름차순 한다

 

 

추가

num.sort(key=lambda x: (-x[1], -x[0]))

 - 를 이용하여 내림차순 할 수 있다

 

 

참고

 

Sorting HOW TO

https://docs.python.org/3/howto/sorting.html

 

2차원 list 정렬

https://velog.io/@therealjamesjung/Python-2%EC%B0%A8%EC%9B%90-list-%EC%A0%95%EB%A0%AC

 

Lambda로 정렬

https://www.delftstack.com/ko/howto/python/lambda-functions-in-sort/

 

728x90
반응형
Share Link
reply