[코딩테스트/백준/Python]좌표 정렬하기 2

🔍좌표 정렬하기 2

<문제>

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

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

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

💡생각

1. N을 입력받는다.
2. for문을 돌리면서 2차원 배열 형식으로 좌표값을 입력받는다.
3. sort()함수를 이용, key=lambda함수를 이용해서 y좌표 기준으로 정렬 후 y값이 같으면 x값을 비교하여 오름차순으로 정렬한다.

💡테스트 통과한 코드

N = int(input())
def main(N):
    num = []
    for i in range(0, N):
        num.append(list(map(int, input().split())))
    num.sort(key=lambda x: (x[1], x[0]))

    for x,y in num:
        print(x,y)
        
main(N)

✅2차원 배열 정렬

1. 0번째 인덱스에 대해서 오름차순으로 정렬, 동일한 값일 경우 내림차순으로 재정렬

   lst = [[2, 1], [3, 4], [1, 2], [1, 3], [3, 2]]
   lst.sort(key=lambda x: (x[0], -x[1]))
   print(lst)
   # [[1, 3], [1, 2], [2, 1], [3, 4], [3, 2]]

2. 1번째 인덱스에 대해서 오름차순으로 정렬, 동일한 값일 경우 오름차순으로 재정렬

   lst = [[2, 1], [3, 4], [1, 2], [1, 3], [3, 2]]
   lst.sort(key=lambda x: (x[1], x[0]))
   print(lst)
   # [[2, 1], [1, 2], [3, 2], [1, 3], [3, 4]]

🔗참고
https://haesoo9410.tistory.com/193

🔗백준 - 좌표 정렬하기 2
https://www.acmicpc.net/problem/11651