Programming/프로그래머스

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

pental 2020. 10. 9. 00:39

def solution(arr, divisor):
    result = []
    count = 0
    Fail = [-1]
    for i in arr :
        if i % divisor == 0 :
            result.append(i)
        else :
            count += 1
    print(count)
    if len(arr) == count :
        return Fail
    return sorted(result)

먼저 원소중 divisor로 나눠지는 것들을 result 리스트에 담고, 그렇지 않다면 count의 개수를 증가시킨다.

그후, 원소의 개수와 count의 개수가 같다면, 나누어지는것들이 없다는 뜻이기에 -1을 반환한다.