[프로그래머스] (Python) - 소수 찾기

문제 분류 : 완전탐색 Lv.2

 

Code(python)

from itertools import permutations as pm

def 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

def check(numbers) :
    numbers = list(numbers)
    cnt = 0
    for i in numbers :
        if prime(i) :
            cnt += 1
    return cnt

def get_permutations(numbers):
    max_len = len(numbers)
    numbers = list(numbers)
    int_nb = list(map(int, numbers))
    if max_len < 2 :
        numbers = map(int, numbers)
        return check(set(numbers))
    else :
        for i in range(2, max_len + 1):
            temp = list(map("".join, (pm(numbers, i))))
            temp = list(map(int, temp))
            int_nb.extend(temp)
    return check(set(int_nb))

def solution(numbers):
    return    get_permutations(numbers)

 

코드 설명

 

prime(n) : n이 소수인지 판별하는 함수

check(numbers): numbers의 원소들을 prime(n) 에 넣고, True면 cnt를 1씩 더해준다.

get_permutations(numbers) : itertools의 permutation 메서드 사용. numbers 로 만들 수 있는 모든 조합의 숫자를 생성한다. ( set과 map을 이용하여 중복을 제거하고, int형으로 conversion했다.)