지원연구소 공식 블로그

[PROGRAMMERS] 소인수분해 (LV. 0) - Python 본문

코딩테스트/프로그래머스

[PROGRAMMERS] 소인수분해 (LV. 0) - Python

지원연구소 2024. 5. 2. 00:13

[ 문제 ]

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

 

 

 

[ 제한 사항 ]

  • 2 ≤ n ≤ 10,000

 

 

[ 입출력 예 ]

n result
12 [2, 3]
17 [17]
420 [2, 3, 5, 7]

 

< 입출력 예 #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. 소수를 입력할 빈 리스트 answer 선언
2. 2부터 n까지 반복 각 값들은 i
3. 반복문 내에서 소수를 판별하기 위한 변수 count 선언
4. i가 소수인지 판별하는 반복문 실행
5. i가 소수이고 n의 약수일 경우 answer 변수에 i 추가
6. 모든 반복이 종료되면 리스트 answer 반환

 

 

 

[ solution.py ]

def solution(n):
    answer = []
    
    for i in range(2, n+1):
        count = 0
        
        for j in range(2, i):
            if i % j == 0:
                count = 1
                break
                
        if count == 0 and n % i == 0:
            answer.append(i)
        
    return answer

 

 

 

> 🌻피드백은 언제나 환영입니다🌻 <