백준 11650번 : 좌표 정렬하기 (python)

 

 

 

 

문제를 통해서 알 수 있는 점 :

1. 2차원 평면 위의 점 N개 → 2차원 배열을 사용해볼까?

2. 좌표를 x좌표가 증가하는 순으로

3. x좌표가 같으면, y좌표가 증가하는 순으로 

4. 정렬 → sorted(), sort() 함수 생각할 수 있음

 

필요한 입력값 :

1. 첫째 줄 : 점의 개수 N

2. 둘째 줄 ~ : i번점의 위치 xi와 yi (위치가 같은 두 점은 없다)

     → 둘째 줄부터 입력 값을 받는 것이므로 반복문 돌려야됨 파악 가능

 

출력값 :

1. 첫째 줄 ~ N개의 줄에 점을 정렬한 결과

→ (한 줄로 배열을 보여주는 것이 아니고)

      각 i번째 배열의 요소를 차례대로 출력해야함을 보고 이 또한 반복문을 돌려서 출력을 해야하는구나 생각할 수 있음

 

 

 

👉 파악한 것을 바탕으로 코드를 작성해보자!

STEP 1 >

많은 값을 입력 받고 있으므로

시간 초과 문제를 해결하기 위해서,

아래와 같이 import sys 를 통해서 input()을 받는 것을 추천한다.

import sys

N = int(sys.stdin.readline().rstrip())

x, y = map(int, sys.stdin.readline().rstrip().split())

 

STEP 2 >

이때, xi와 yi는 반복문을 통해서 입력을 받아야 되기 때문에

N의 크기 만큼의 for문을 사용해서 xi와 yi의 값을 입력 받는다.

import sys

N = int(sys.stdin.readline().rstrip())

for i in range(N):
    [x, y] = map(int, input().split())

 

STEP 3 >

xi와 yi의 점들은 2차원 배열 안에 있을 요소들이므로

array 배열을 하나 만들고

각각의 xi와 yi 점들을 하나의 요소로 append해준다. 

import sys

N = int(sys.stdin.readline().rstrip())

array = []
for i in range(N):
    x, y = map(int, sys.stdin.readline().rstrip().split())
    array.append([x,y])

 

STEP 4 >

이후, sorted() 함수를 이용해서 배열 array를

x좌표가 증가하는 순으로 x좌표가 같으면, y좌표가 증가하는 순인

배열 s_array로 만들어준다.

import sys

N = int(sys.stdin.readline().rstrip())

array = []
for i in range(N):
    x, y = map(int, sys.stdin.readline().rstrip().split())
    array.append([x,y])

s_array = sorted(array)

 

STEP 5 >

마지막으로, 주어진 출력값처럼 출력되게

for문을 N의 크기만큼 돌려서

s_array 배열에 저장된 i번째의 xi와 yi 값을 출력해주면 된다. 

import sys

N = int(sys.stdin.readline().rstrip())

array = []
for i in range(N):
    x, y = map(int, sys.stdin.readline().rstrip().split())
    array.append([x,y])

s_array = sorted(array)

for i in range(N):
    print(s_array[i][0], s_array[i][1])

 

 

 

👉 최종 코드는 이러하다

import sys

N = int(sys.stdin.readline().rstrip())

array = []
for i in range(N):
    x, y = map(int, sys.stdin.readline().rstrip().split())
    array.append([x,y])

s_array = sorted(array)

for i in range(N):
    print(s_array[i][0], s_array[i][1])

 

 

 

👉 Q&A

📌 왜 sorted() 함수에서 lambda를 사용하지 않았냐?

#ex1) 
array.sort(key = lambda x: (x[0],x[1]))

#ex2)
arr = sorted(arr, key= lambda y: y[1])
arr = sorted(arr, key= lambda x: x[0])

#ex3) 
arr.sort(key=lambda a : (a[0], a[1]))

#my_code
s_array = sorted(array)

 

 

 

👉 알면 좋은 코드들

📍이렇게 미리 쓰기만 하면, 그 이후에 바로 input() 함수를 사용할 수 있음. 

import sys

input = sys.stdin.readline

N = int(input())

#N = int(sys.stdin.readline().rstrip())

(참고) https://puleugo.tistory.com/42

 

📍이런식으로 코드를 단축 시킬 수 있음. 

array = []
for _ in range(N) :
    array.append(list(map(int, input().split())))
    
#my_code
#array = []
#for i in range(N):
#    x, y = map(int, sys.stdin.readline().rstrip().split())
#    array.append([x,y])

 

📍*를 사용하면, 리스트 안의 요소들만 꺼내서 한 줄에 한 번에 출력할 수 있다.

#ex1)
for x in arr:
    print(*x)
    
#ex2)    
for i in range(0, N):
    print(*arr[i])

#my_code
#for i in range(N):
#    print(s_array[i][0], s_array[i][1])

(참고) https://yeomss.tistory.com/160