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 |