백준 11651 번 문제 풀이 Python 파이썬 :좌표 정렬하기 2

#81. 백준 11651번 문제 풀이: 좌표 정렬하기 2

2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오. 입력: 첫째 줄에 점의 개수 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

문제풀이+해설

좌표와 같은 2차원 배열형태의 배열을 소팅하는 문제이다.
언어에 따라 2차원 배열 소팅하는 기능을 기본적으로 제공하는 언어가 있고, 그렇치 않은 언어들이 있다.파이썬에서는 소팅을 할 때, 비교하는 함수를 넣어줄 수 있다.
여기서는 lambda 식으로 써서 넣었다.

sort(key=lambda p: (p[1], p[0])) 
소팅할 때 p[1]을 우선으로, 같으면 p[0]을 기준으로 소팅하라는 의미이다.
즉, y값을 우선적으로 소팅하고 같으면 x값을 기준으로 소팅하라는 의미이다.

그리고, 값을 입력받을 때 input() 함수를 쓰면 시간초과가 발생하므로 속도가 좀 더 빠른 sys.stdin.readline() 함수를 사용했다.

code sol.

코드편집/실행

import sys
N = int(sys.stdin.readline())
#좌표 개수를 입력받아 정수형으로 변환해서 대입
ps = [] # 좌표값을 저장할 리스트

for _ in range(N): # 주어진 개수만큼 반복
    ps.append(list(map(int, sys.stdin.readline().split())))
    # 좌표값을 받아서 공백으로 나누고 정수로 변환해서 리스트로 만들어 추가

ps.sort(key=lambda p: (p[1], p[0])) 
# y좌표(p[1]) 우선, 그 다음 x좌표(p[0])우선으로 소팅

for p in ps: # 소팅된 리스트에서 하나씩 꺼내서
    print(p[0], p[1]) # x, y 좌표값 하나씩 출력