Python/백준

11651_좌표 정렬하기2

728x90

문제는 더보기!

 

더보기

문제

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

입력

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

출력

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

예제 입력 1 복사

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

예제 출력 1 복사

1 -1
1 2
2 2
3 3
0 4

 

저번 좌표정렬하는것과 거의 유사한 문제이다. 문제를 딱 보고 그냥 저번에 a,b만 바꾸면 되는거 아니야? 라는 생각이 들었다.

 

import sys
N = int(sys.stdin.readline())
lst=[]
for _ in range(N):
    a,b = list(map(int,sys.stdin.readline().split()))
    lst.append([b,a])

lst.sort()

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

끝!

 

이라고하기엔 찝찝하다. 항목이 3개들어올때나 특정 항목만 내림차순하고 싶을땐 어떻게 할수가 없다. 그래서 sort함수를 조금 공부해 보았다. key와 lambda를 이용하면 [a,b] 이렇게 온것중에서 뒤에것만 정렬하는것이 가능하다.

import sys
N = int(sys.stdin.readline())
lst=[]
for _ in range(N):
    a,b = list(map(int,sys.stdin.readline().split()))
    lst.append([a,b])

lst.sort(key = lambda x : x[1])

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

사용방법은 디게 간단했다. 즉, 정렬하는 key값을 x로 정해주는 방식이라고 생각하면 될 것 같다.

728x90

'Python > 백준' 카테고리의 다른 글

10814_나이순 정렬  (0) 2021.12.07
1181_단어 정렬  (0) 2021.12.05
11650_좌표 정렬하기  (0) 2021.12.03
1427_소트인사이드  (0) 2021.12.03
2108_통계학  (0) 2021.12.01