본문 바로가기

알고리즘

[프로그래머스] 없는 숫자 더하기(python)

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

📌 문제 설명

0부터 9까지의 숫자 중 일부가 들어있는 정수 배열 numbers가 매개변수로 주어집니다. numbers에서 찾을 수 없는 0부터 9까지의 숫자를 모두 찾아 더한 수를 return 하도록 solution 함수를 완성해주세요.


✔️제한사항
  • 1 ≤ numbers의 길이 ≤ 9
    • 0 ≤ numbers의 모든 원소 ≤ 9
    • numbers의 모든 원소는 서로 다릅니다.

 

✔️입출력 예
numbers result
[1,2,3,4,6,7,8,0] 14
[5,8,4,0,6,7,9] 6

✔️입출력 예 설명

입출력 예 #1

  • 5, 9가 numbers에 없으므로, 5 + 9 = 14를 return 해야 합니다.

입출력 예 #2

  • 1, 2, 3이 numbers에 없으므로, 1 + 2 + 3 = 6을 return 해야 합니다.

🗒 풀이

def solution(numbers):
    return sum([i for i in range(10) if i not in numbers])

0부터 9까지 for문을 탐색하며, 입력받은 numbers 리스트에 없는 경우만 남기고, 그것들을 sum 함수로 더한 것을 반환한다.


🤔 후기

처음에는 numbers에 없는 리스트를 조회해서 그것들만 더하면 되겠다고 생각하고 접근했다.

for문을 작성하고 있었는데, 잘하면 한 줄로 작성할 수 있겠다 싶어 list comprehension을 사용했다.

나름 뿌듯해했으나, 더 괜찮은 코드가 있었다.

return 45 - sum(numbers)

0부터 9까지의 합인 45에서 numbers의 합을 빼면 더 쉽게 해결할 수 있는 것이었다. 쩝 ㅎ