백준 11651번 - 좌표 정렬하기 2 / Python

문제: https://www.acmicpc.net/problem/11651

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

input = sys.stdin.readline

N = int(input())
points = []
for _ in range(N):
    x, y = map(int, input().split())
    points.append((x, y))

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

for x, y in points:
    sys.stdout.write(f'{x} {y}\n')

 

---

풀이

import sys

input = sys.stdin.readline

N = int(input())
points = []
for _ in range(N):
    x, y = map(int, input().split())
    points.append((x, y))

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

for x, y in points:
    sys.stdout.write(f'{x} {y}\n')

 기존 좌표 정렬하기 문제의 응용 버전입니다. y, x 순서로 정렬을 하기 때문에 key를 이용해서 정렬을 해주면 됩니다.