[코딩테스트 입문] 소인수분해

문제 설명

소인수분해란 어떤 수를 소수들의 곱으로 표현하는 것입니다. 예를 들어 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합니다.

---

 

머리가 굳었는지 이런 간단한 문제도 머리를 써서 풀지 못하겠다. 다른 사람의 답을 보니 바로 알겠다.

우리가 아는 그 소인수분해, 그것만 잘 알면 되는 문제였다.

 

소인수분해하는 함수를 따로 만들어서 활용했다. 빈 리스트와 소인수인 d를 정의하고, d가 n에 나누어떨어지면 리스트에 추가하는 방식이다. 더이상 나누어떨어지지 않으면 d의 수를 높인다. 그리고 마지막에 sorted와 set을 통해 리스트를 정리한다. 

def factorization(n):
    lst, d = [], 2
    while n > 1:
        if n%d == 0:
            lst.append(d)
            n = n//d
        else:
            d += 1
    return lst

def solution(n):
    answer = sorted(set(factorization(n)))
    return answer

 

끝 !