백준 #11650

기억할 것

1. input을 sys.stdin.readline()로 받을 것

2. box.sort(key=lambda x: (x[0], x[1])) 

3. print('\n'.join([f"{x[0]} {x[1]}" for x in box]))

print('\n'.join([f"{x[0]} {x[1]}" for x in box]))는 리스트 컴프리헨션을 사용하여 box 리스트의 각 요소를 형식에 맞추어 문자열로 변환하고, 그 문자열들을 줄바꿈 문자('\n')로 연결하여 출력하는 코드이다.

해당 코드는 box 리스트의 요소를 반복하면서 각 요소를 문자열로 변환하고, 이들을 리스트로 모아서 [f"{x[0]} {x[1]}" for x in box]로 만든 후, '\n'.join(...)을 사용하여 리스트의 요소들을 줄바꿈 문자('\n')로 연결하여 하나의 문자열로 만든 후, 이를 print() 함수로 출력한다. 

 

문제

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

입력

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

출력

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

예제 입력 1 복사

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

예제 출력 1 복사

1 -1
1 1
2 2
3 3
3 4

내가 쓴 시간초과 나는 코드

import sys
num = int(input())
box = []
for i in range(num):
    box.append(list(map(int, sys.stdin.readline().split())))

for j in range(len(box)):
    box.sort(key = lambda x:x[0])
    box.sort(key = lambda x:x[1])


for k in range(len(box)):
    print(box[k][0], box[k][1])

정답 코드

import sys

num = int(input())
box = []

for i in range(num):
    box.append(list(map(int, sys.stdin.readline().split())))

box.sort(key=lambda x: (x[0], x[1]))

print('\n'.join([f"{x[0]} {x[1]}" for x in box]))

+) 여담

백준을 슬슬 접어야 하나 고민이다. 

데이터 사이언스를 하는 것에 있어서 시간 초과가 나는 것을 계속 고려하는 것이 의미있는 행동인지 모르겠다. 

pandas나 numpy도 사용을 못하는 것도 아쉽긴 하다.