[프로그래머스 / Python] 가까운 수

문제 출처

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

---

문제

정수 배열 array와 정수 n이 매개변수로 주어질 때, array에 들어있는 정수 중 n과 가장 가까운 수를 return 하도록 solution 함수를 완성해주세요.

---

포인트

이 문제의 핵심은 두 가지입니다. 첫 번째는 주어진 n과의 차이(절대값)가 가장 작은 원소를 찾는 것이고, 두 번째는 그러한 원소가 여러 개일 경우 더 작은 값을 반환하는 것입니다.

---

내 답안

def solution(array, n):
    # 주어진 배열에서 n과 가장 가까운 수 찾기
    # 거리가 같은 경우 더 작은 수 반환
    return min(array, key=lambda x: (abs(x - n), x)) 

---

결론 및 느낀점

def solution(array, n):
    return min(array, key=lambda x: abs(x - n))

간단한 문제라고 생각하고 코드를 제출 했더니 실패하는 테스트들이 발생했다. 다시 문제를 보니 제한사항에 가장 가까운 수가 여러 개일 경우 더 작은 수를 return 합니다. 라는 조건이 있었다. 아마 테스트들 중에 abs(x - n) 값이 같은 숫자들이 있었나보다. 문제 조건을 잘 살펴보고, 파이썬에서 제공하는 편리한 함수들을 잘 활용한다면 더 깔끔하고 효율적으로 문제를 풀 수 있을 것 같다.