[백준] 11651 : 좌표 정렬하기 2 (Python/파이썬)

문제 링크

https://www.acmicpc.net/problem/11651

문제

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

- 시간 제한 : 1초
- 메모리 제한 : 256MB

입력

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

출력

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

예제

입력

5
0 4
1 2
1 -1
2 2
3 3

출력

1 -1
1 2
2 2
3 3
0 4

---

정답 및 풀이

import sys

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

for i in arr:
    print(*i)

1. import sys
   • 입력이 많을 때 속도를 높이기 위해 sys.stdin.readline() 사용한다.
2. n = int(sys.stdin.readline())
   • 첫 번째 줄에서 좌표의 개수 n을 입력받는다.
3. arr.append(list(map(int, sys.stdin.readline().split())))
   • 한 줄에 (x, y) 형태로 주어진 좌표를 정수 리스트로 저장한다.
   • arr = [[x1, y1], [x2, y2], ...] 형태
4. arr.sort(key=lambda x: (x[1], x[0]))
   • lambda를 이용해 정렬 기준(key) 설정한다.
   • x[1]: y좌표 기준으로 오름차순 정렬
   • x[0]: y좌표가 같을 경우 x좌표 기준으로 정렬
5. print(*i)
   • 리스트 언패킹()을 이용해 공백으로 구분해 출력한다.

---

새롭게 배운 내용 및 느낀점

• sort(key = 기준함수)
  • 리스트를 정렬할 때 기준을 지정할 수 있음
• lambda
  • lambda 매개변수: 표현식
  • 함수를 한 줄로 간단히 표현할 때 사용
  • lambda x : x[1] - 입력받은 x 의 두 번째 요소를 반환