[Silver V] 좌표 정렬하기 - 11650

문제 링크

성능 요약

메모리: 59988 KB, 시간: 420 ms

분류

정렬

문제 설명

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 _ in range(N):
    new = list(map(int, sys.stdin.readline().split()))
    array.append(new)

sorted_list = sorted(array, key=lambda x : (x[0], x[1]))

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

코드 리뷰

이 문제는 x, y좌표를 받아서 정렬하는 문제이다. 파이썬을 사용하면 매우 쉽게 정렬할 수 있는데, sorted_list = sorted(array, key=lambda x : (x[0], x[1])) 이 코드의 의미는 리스트의 첫 번째 원소를 기준으로 정렬하되, 그 값이 같다면 두 번째 원소를 고려한다는 의미이다. 파이썬은 이와 같이 정렬에 관해서 뿐 아니라 매우 많은 내장 함수들이 존재하기 때문에 알고리즘 시험에 있어서는 파이썬이 조금이라도 시간이 단축되거나 유리하지 않나 하는 생각을 한다.

댓글남기기