[완전탐색] lv.2 소수 찾기 | Python 파이썬

문제 :

 

 

코드: 

from itertools import permutations

# 소수 판별 함수
def is_prime_num(n):
    if n <= 1:
        return False
    
    for i in range(2, int(n**(1/2))+1):
        if n % i == 0:
            return False
    return True
    
def solution(numbers):
    num = list(numbers)
    allnum = set()
    
    # 순열과 set사용해서 모든 조합 구하기
    for i in range(1, len(numbers)+1):
        permutationList = permutations(num, i)
        for per in permutationList:
            allnum.add(int(''.join(per)))
    
    cnt = 0
    
    # 소수 판별
    for n in allnum:
        if is_prime_num(n):
            cnt+=1
            
    return cnt

 

 

풀이 : 

1. 소수를 구할때는 에라토스테네스의 채 사용해서 루트+1 까지만 소수인지 판별하면 된다.

2. permutations(iterable, r)을 사용하면 iterable의 요소들의 길이 r인 순열로 반환한다. 이 문제에서는 모든 조합을 구하기 위해 for문을 사용하여 길이가 i로 바뀌도록 했다.

3. 중복 제거하기 위해 set에 넣기!

4. ''.join 활용