백준 11650번. 좌표 정렬하기 (Python / 파이썬)

문제

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

입력

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

출력

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

---

문제 풀이

2차원 평면 위의 점이 N개 주어졌을 때

x좌표가 증가하는 순으로,

x좌표가 같으면 y좌표가 증가하는 순으로 

정렬하는 문제입니다.

 

sort() 함수는 key를 사용할 수 있는데

key는 어떤 원소를 기준으로 정렬할지 지정할 수 있습니다.

이때, 만약 key를 여러개 지정하면

key는 순서대로 우선순위를 가집니다.

 

즉, sort(key = lambda x: (x[0], x[1])이면

x[0]을 우선적으로 기준으로 하여 정렬하고,

그 다음 x[1]을 기준으로 정렬해줍니다.

 

따라서 이를 이용하여 x좌표를 우선으로 먼저 정렬한 뒤

y좌표가 정렬되로록 해주었습니다.

---

My Code

n = int(input()) # 점의 개수

pos = []
for _ in range(n):
    x, y = map(int, input().split()) # x좌표, y좌표
    pos.append((x, y)) # list에 삽입

# x좌표를 기준으로 정렬, x좌표가 같으면 y좌표를 기준으로 정렬
pos.sort(key = lambda p: (p[0], p[1])) 

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