본문 바로가기
코테 준비/프로그래머스

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

by 귤이두번 2022. 2. 18.

 

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


나누어 떨어지는 수들만 answer 리스트에 추가

만약 아무것도 추가되지 않아서 리스트 길이가 0이면 -1추가

 

마지막으로 정렬된 리스트 리턴

댓글