나누어 떨어지는 숫자 배열

2019. 12. 4. 00:23개발노트

 

arr 요소를 divisor로 나누고 나머지가 0인 값들만 리스트에 append 시켰습니다.

나누어 떨어지지 않아서 리스트가 비어있다면 -1을 append 시킵니다.

 

return값은 오름차순으로 정렬하기 때문에 sorted를 사용했습니다.

 

def solution(arr, divisor):
    answer = []
    temp = []
    for a in arr:
        if a % divisor is 0:
            temp.append(a)
    if not temp:
        temp.append(-1)
    temp = sorted(temp)
    answer = temp
    return answer