[BOJ 20949, Python 3] 효정과 새 모니터

https://www.acmicpc.net/problem/20949

20949번: 효정과 새 모니터

 

풀이

간단한 정렬 문제 입니다. sort안에 key와 lambda를 사용하여 정렬을 해주면 됩니다.

문제에서 가로 픽셀과 세로 픽셀만 주어졌으므로 (가로 픽셀)^2 + (세로 픽셀)^2 기준으로 정렬해도 무방합니다.

 

 

코드

n = int(input())
arr = sorted([[*map(int, input().split())] + [_+1] for _ in range(n)], key=lambda x:(x[0]**2+x[1]**2), reverse=True)
for i in range(n): print(arr[i][2])