문제

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에 비하면 느리다는게... 참 아쉽다.

 

 

코드가 이상하거나, 이해되지 않는 부분이 있다면 댓글 남겨주세요!

+ 더 좋은 풀이 방법이 있다면 알려주세요 :)

+ Recent posts