[알고리즘 문제풀이] 나누어 떨어지는 숫자 배열

ljkgb·2021년 3월 4일
0

알고리즘 문제풀이

✔문제

프로그래머스

✔내가 작성한 코드

def solution(arr, divisor):
  answer = []
  for i in arr:
    if i % divisor == 0:
      answer.append(i)
    else:
      continue
  if answer == []:
    answer.append(-1)
  return sorted(answer)

✔답안 예

def solution(arr, divisor): return sorted([n for n in arr if n%divisor == 0]) or [-1]

✔Review

이론적으로 이렇게 함수를 쓴다는건 배웠었지만 적용은 못했는데 앞으로 간단한 함수는 이렇게 사용하도록 해봐야겠다!

profile
🐹

0개의 댓글