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

문제

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

입력

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

출력

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

답안

# lambda를 이용해 0,1순으로 정렬해줬다.
t = int(input())
arr = []
for _ in range(t):
    xy = list(map(int, input().split()))
    arr.append(xy)
arr.sort(key = lambda x: (x[0], x[1]))
for i in arr:
    print(*i)
    
# sort 메서드를 이용해도 된다...
t = int(input())
arr = []
for _ in range(t):
    lst = list(map(int, input().split()))
    arr.append(lst)
arr.sort()
for i in arr:
    print(*i)

후기

0, 1 순으로 정렬은 알아서 해주기 때문에 sort메서드로도 사용이 가능하다.