[Alogorithm] [Python] 백준/BOJ - 11651_좌표 정렬하기2

11651 좌표 정렬하기 2

---

문제

---

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

입력

---

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

출력

---

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

1차 제출 코드

---

import sys
N = int(sys.stdin.readline())
arr = []
for _ in range(N):
   arr.append(list(map(int, sys.stdin.readline().split(' '))))
arr.sort(key=lambda x: (x[1], x[0]))
for data in arr:
   print(data[0], data[1])

이 코드도 정상적으로 답으로 처리되었다.

근데, 뭔가 더 줄일수 있을것 같아서 최대한 간결하게 줄여서 다시 제출해보았다!

2차 제출 코드

---

import sys
N = int(sys.stdin.readline())
arr = [list(map(int, sys.stdin.readline().split())) for _ in range(N)]
arr.sort(key=lambda x : (x[1], x[0]))
for i in arr:
   print(i[0], i[1])

1차 제출 코드의 3~5번 라인을 위 예시에서는 3번 줄 하나로 해결했다.

배열을 선언과 N번 반복하며 동시에 입력을 받으며 이를 int형으로 변환했다.

코드 설명

---

Line 1 : sys.stdin.readline() 사용을 위한 sys module 추가

Line 2 : N값 입력받기

Line 3 : N번 반복하며 좌표를 입력받아 nested list 방식으로 데이터 추가

Line 4 : y좌표를 우선으로 하며 만일 겹칠시 x좌표를 기준으로 정렬하기 위해 lambda식을 활용한 key설정 이후 arr리스트 정렬

Line 5~6 : 정렬된 배열 원소 출력