[프로그래머스] Lv.0 - 소인수분해.py

문제

• 소인수분해란 어떤 수를 소수들의 곱으로 표현하는 것입니다. 예를 들어 12를 소인수 분해하면 2 * 2 * 3 으로 나타낼 수 있습니다. 따라서 12의 소인수는 2와 3입니다. 자연수 n이 매개변수로 주어질 때 n의 소인수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 2 ≤ n ≤ 10,000

입출력 예시

• 입출력 예 1

  • 12를 소인수분해하면 2 * 2 * 3 입니다. 따라서 [2, 3]을 return합니다.

• 입출력 예 #2

  • 17은 소수입니다. 따라서 [17]을 return 해야 합니다.

• 입출력 예 #3

  • 420을 소인수분해하면 2 * 2 * 3 * 5 * 7 입니다. 따라서 [2, 3, 5, 7]을 return합니다.

코드 1

def solution(n):
    answer = []
    for num in range(2, n+1):
        while(n / num == int(n/num)):  
            answer.append(num)
            n = n / num
    answer = sorted(list(set(answer)))
    return answer

코드 2

def solution(n):
    answer = []
    for num in range(2, int(n/2)+1):
        while(n / num == int(n/num)):  
            answer.append(num)
            n = n / num
    if len(answer) == 0:
        answer.append(n)
    answer = sorted(list(set(answer)))
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges