본문 바로가기

알고리즘 삽질장

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

반응형


문제설명

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

 

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

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

programmers.co.kr

사고과정

  • 단순한 구현 문제였다. 배열 원소 하나씩 돌면서 divisor과 나누어떨어지는지 확인하면 됨. 단, 마지막에 나누어떨어지는 숫자를 append 한 리스트 길이가 0이면 -1을 반환하는 처리만 추가해주면 된다.

풀이

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