[프로그래머스] 나누어 떨어지는 숫자 배열 - python

문제

• 프로그래머스 연습문제- 나누어 떨어지는 숫자 배열
• https://programmers.co.kr/learn/courses/30/lessons/12910

풀이

주어진 배열을 divisor로 나누어 나머지가 0인 값만 ans배열에 넣는다.

만약 ans배열이 비어있다면, -1을 넣고 리턴한다.

코드

def solution(arr, divisor):
    ans=list()
    for i in range(len(arr)):
        if arr[i]%divisor == 0:
            ans.append(arr[i])
    if ans == []:
        ans=[-1]
    ans.sort()
    return ans