Lv2. 소수 찾기

코딩테스트 연습 > 소수 찾기

1. 풀이 설명

2. 나의 풀이

from itertools import permutations

def solution(numbers):
    result = []
    for i in range(1, len(numbers)+1):
        for tp in permutations(numbers, i):
            num = int(''.join(map(str, tp)))
            if is_prime(num):
                result.append(num)
    return len(set(result))


def is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

3. 배운점

python

1. 순열을 구해주는 라이브러리를 알게 되었다: from itertools import permutations

from itertools import permutations

permutation(list, n) # list 를 n개씩 묶어주는 리스트를 반환

참고