[파이썬/Python] 백준 11650번 좌표 정렬하기

[파이썬/Python] 백준 11650번 좌표 정렬하기

1920번: 수 찾기

---

문제

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

풀이

import sys
input = sys.stdin.readline

# N개의 점을 입력 받아 2차원 리스트 형태로 저장
lst = [list(map(int, input().split())) for _ in range(int(input()))]

# x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 후 출력
for x, y in sorted(lst):
    print(x, y)

이 코드의 시간 복잡도는 입력 받은 N개의 점을 정렬하는 부분에서 O(N log N)이다. 따라서 전체적인 시간 복잡도는 O(N log N)이다.

 

sort 는 리스트의 순서를 보장하지만 이는 키 값으로 람다를 사용할 경우이고 디폴트는 전체 정렬이므로 혼동하지 않도록 주의한다.

참고자료

(파이썬) 백준 알고리즘 - 11650번, 11651번 / sort(), lambda 함수

반응형

저작자표시 (새창열림)