백준 11650번 (python) - 좌표 정렬하기

좌표 정렬하기

백준 > 단계별로 풀어보기 > 정렬

문제

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

입력

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

출력

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

풀이

import sys

N = int(sys.stdin.readline())

array = []

for i in range(N):
    array.append(list(map(int, sys.stdin.readline().split())))

array = sorted(array)

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

파이썬이 얼마나 부당한 언어인지 깨달았다.

우선, list와 map 함수를 통해 array에 두 개씩 입력을 받고 append를해, 좌표평면 형태를 만든다.

이 때, 이차원 배열 형태로 저장되는데, sorted함수를 통해 2차원 배열이든 1차원 배열이든 상관없이 오름차순으로 결과를 돌려받는다.

그 외에 sorted함수 파라미터 옵션을 통해 내림차순도 가능하다.

뿐만 아니라, key value 파라미터를 통해 다양한 기준을 통해 다차원(아마도 2차원) 배열을 정렬할 수 있다.