본문 바로가기
알고리즘/프로그래머스 PS

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

by 빛밤하늘 2021. 7. 6.
반응형

밑의 링크는 프로그래머스에서의 문제 링크입니다.

 

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

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

programmers.co.kr

 

 

 

 

 

 

 

 

파이썬 코드와 결과입니다.

# 자연수를 담은 배열 arr, 자연수인 divisor를 매개변수로 받습니다.
def solution(arr, divisor):
    # 정답들을 저장할 리스트 변수를 선언합니다.
    answer = []

    # arr에 있는 자연수들을 하나씩 반복합니다.
    for num in arr:
        # 현재 숫자가 divisor로 나누어 떨어진다면
        if num % divisor == 0:
            # answer에 현재 숫자를 넣어줍니다.
            answer.append(num)

    # answer가 빈 리스트라면
    if answer == []:
        # answer를 -1 하나가 담긴 리스트로 만들어줍니다.
        answer = [-1]
    # answer가 빈 리스트가 아니라면
    else:
        # answer의 값들을 오름차순으로 정렬합니다.
        answer.sort()

    # answer를 반환합니다.
    return answer

결과

 

 

 

※ 궁금한 부분, 이상한 점 및 오타는 댓글에 부탁드립니다.

※ 더 효율적이고 빠른 정답을 환영합니다.

반응형

댓글