[프로그래머스] 소수 찾기/ 파이썬(python)

Jiwon_C 2021. 5. 28. 02:31

# 문제 링크

https://programmers.co.kr/learn/courses/30/lessons/42839

# Soultion

순열을 이용하여 모든 경우의 수를 리스트 num에 넣어주었다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

from itertools import permutations
def solution(numbers):
    answer = 0
    num = []
    for i in range(1,len(numbers)+1):
        for j in list(permutations(numbers,i)):
            num.append(int(''.join(j)))
 
    for i in list(set(num)): #중복제거
        if i==2:
            answer+=1
        for j in range(2,i):
            if i%j==0: #나누어떨어지면 -> 소수아님
                break
            elif j==i-1: #소수이면 answer+=1
                answer+=1
                
    return answer

cs

5~7. 순열을 사용하여 모든 경우의 수 저장

9. 중복된 숫자 제거를 위해 set 사용

10. 숫자가 2인경우는 따로 처리하여 더해준다.

12. 2부터 i-1까지 숫자를 비교하여 소수찾기