프로그래머스(Python) 풀이/Level.1

프로그래머스(Python) Level.1 나누어 떨어지는 숫자 배열 풀이

개발윗미 2022. 3. 2. 14:52

Python으로 구현한 12910번 나누어 떨어지는 숫자 배열 문제 풀이입니다.

 

https://programmers.co.kr/learn/courses/30/lessons/12910

 

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요. divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하

programmers.co.kr


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

 

1. 반복문을 통해 arr리스트 내 요소들을 하나씩 확인하여 divisor로 나누어 떨어질 경우 answer리스트에 값을 추가한다.

 

2. 반복문 작업이 끝난 후 answer리스트의 길이가 0일 경우 divisor로 나누어 떨어지는 요소가 하나도 없으므로 [-1]을 반환한다.

 

3. 그렇지 않을 경우 answer리스트를 정렬하여 반환한다.