본문 바로가기

Algorithm/프로그래머스 연습 문제

프로그래머스 / 코딩 테스트 / 나누어 떨어지는 숫자 배열

문제 설명

 

 

문제 해결

 

리스트의 각 원소를 divisor로 나머지 연산한 값이 0인지 확인하고 원소들을 다시 모아 리스트를 구성하면 된다.

comprehension 기법으로 리스트를 재구성해주고 sorted 함수를 이용하여 오름차순으로 정렬해줬다.

재구성한 리스트의 길이가 0일 경우 나누어 떨어지는 원소가 없다는 것이므로 [-1]을 반환한다.

 

def solve(arr, divisor):
    res = sorted([x for x in arr if x % divisor == 0])
    return res if len(res) != 0 else [-1]

def solution(arr, divisor):
    answer = solve(arr,divisor)
    return answer