[백준] #11650 좌표 정렬하기(python)

📌문제

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

백준 11650번 문제

💡Idea

파이썬의 sort함수를 사용하여 key 옵션🔑을 통해 다중 조건을 설정해주면 되는 문제입니다.

sort함수의 경우, key 옵션으로 주어지는 함수의 결과에 따라서 정렬을 합니다.

만약 비교할 요소가 두 개 이상인 경우, 튜플이나 리스트를 통해 순서대로 반환하도록 해주면 됩니다.
그러면 첫 번째 요소를 기준으로 정렬을 한 뒤, 첫 번째 요소가 동일하다면 두 번째 요소를 비교하여 정렬하도록 합니다.

 arr.sort(key=lambda x: (첫 번째 정렬 기준, 두 번째 정렬 기준))

💻코드

import sys
input = sys.stdin.readline

N = int(input())
num = [list(map(int, input().split())) for _ in range(N)]
num.sort(key=lambda x: [x[0], x[1]])

for n in num:
    print(n[0], n[1])

📝코드 설명

변수

• N : 정렬해야 하는 평면 위의 점 개수
• num : 점들의 좌표를 잠은 리스트

좌표 평면 위 점들의 좌표를 [x좌표, y좌표]의 형태로 리스트에 저장한 뒤, x좌표를 정렬의 첫 번째 요소, y좌표를 정렬의 두 번째 요소로 지정합니다.

정렬이 완료되면, 정렬된 리스트를 순서대로 출력합니다.