[CT] Programmers 소인수분해 Python3 ver.

 

[문제 설명]

 

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

 

---

[나의 풀이]

def solution(n):
    answer = []
    div = 2
    
    while div <= n:
        if n % div == 0:
            if div not in answer:
                answer.append(div)
            n //= div
        else : div += 1
    return answer

 

1. div를 2로 초기화한다.

 

2. div가 n보다 작거나 같을 때까지

 

3. 만약 n을 div로 나누었을 떄 나머지가 0이라면 (div가 n의 소인수라면)

 

4. 그리고 이 때 div가 answer 리스트에 존재하지 않는다면

 

5. answer에 div를 추가한다.

 

6. 만약 n을 div로 나누었을 떄 나머지가 존재한다면(div가 n의 소인수가 아니라면)

 

7. div에 1을 더한 후 while문을 계속 돈다.