[python] 가까이있는 수 구하기 - sort, lambda

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120890

프로그래머스

 

나의 풀이

def solution(array, n):
    min_ = 99999
    result = -1
    array.sort()
    
    for i in array:
        if abs(i-n) < min_:
            min_ = abs(n-i)
            result = i
            
    return result

주어진 배열 array에 수들이 들어있고 이 중 n과 가장 가까운 숫자를 골라야한다.

단, 가장 가까운 숫자들이 많을 경우 그 중 가장 작은 수를 골라야 함.

1. array를 정렬

2. for문을 사용해서 가까운 수의 값을 result에 넣음

 

인상적인 풀이

def solution(array, n):
    array.sort(key = lambda x : (abs(x-n), x-n))
    answer = array[0]
    return answer

array를 정렬할 때 "가까운 수들 중 가장 작은 값"을 넣기 위해 lambda가 쓰였다.

sort key 값에 2개 항목을 넣을 수 있다.

첫번째 정렬과 두번째 정렬에 x-n을 넣어주는 이유는 abs(x-n)이 똑같은 경우에 값이 작은게 앞에 있게 하기 위해서이다.