[programmers - python] Lv.1: 나누어 떨어지는 숫자 배열

문제 링크

프로그래머스

문제 풀이

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

나누어 떨어지는 숫자 배열 입출력 테스트

*key point: list comprehension, sorted 함수 활용하여 주어진 조건의 list를 얻는다. 결과가 빈 리스트인 경우에만 [-1] 을 할당한다.

 

*comment: 아래와 같이 or 을 활용하면 빈 리스트인 경우에 대한 리턴을 쉽게 해줄 수 있다.

return sorted([e for e in arr if e % divisor == 0]) or [-1]