문제
2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.
조건 및 입출력
1. 첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.
2. 첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.
코드 (Python 3)
# 정의부
import sys
# 입력부
trys=int(sys.stdin.readline())
num=[];
for i in range(trys):
num.append(list(map(int,sys.stdin.readline().split())));
# 처리부
num.sort(key=lambda x: (x[0],x[1]));
# 출력부
for i in range(trys):
print(*num[i])
풀이 과정
그냥 input()으로 값을 입력받으면 시간 초과가 된다. sys.stdin.readline()으로 값을 받아야 시간 초과가 되지 않는다.
일단 빈 리스트를 하나 만들고, 거기다 x,y값을 또 리스트 형식으로 만들어 저장한다.
그렇게 되면 2차원 배열이 만들어지고, lambda 함수를 이용해 0번째(x값)를 기준으로 정렬하고, 그 다음 1번째(y값)을 비교하여 정렬한다.
그 후 정렬된 리스트를 출력한다.
여담
파이썬이 다 좋은데, 가끔씩 시간 초과가 날 때가 있다.
C에 비하면 느리다는게... 참 아쉽다.
| 코드가 이상하거나, 이해되지 않는 부분이 있다면 댓글 남겨주세요! + 더 좋은 풀이 방법이 있다면 알려주세요 :) |
'Programming > Python' 카테고리의 다른 글
| [백준] 1357번 : 뒤집힌 덧셈 (Python / 파이썬) (0) | 2023.01.30 |
|---|---|
| [백준] 16395번 : 파스칼의 삼각형 (Python / 파이썬) (2) | 2023.01.29 |
| [백준] 8741번 : 이진수 합 (Python / 파이썬) (0) | 2023.01.28 |
| [백준] 1676번 : 팩토리얼 0의 개수 (Python / 파이썬) (0) | 2023.01.27 |
| [백준] 2744번 : 대소문자 바꾸기 (Python / 파이썬) (0) | 2023.01.27 |