본문 바로가기

Problem Solving

[Brute Force] 소수 찾기-프로그래머스(Lv.2)

문제 제목 : 소수 찾기

문제 출처 : 프로그래머스

알고리즘 : Brute Force

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

from itertools import permutations
def makePrime(n):
    prime=[0,0]+[1]*~-n 
    for i in range(int(n**.5)+1):
        if prime[i]:prime[i*i::i]=[0]*(n//i-i+1)
    return prime

def solution(numbers):
    ans = 0; nlen = len(numbers)
    num = [i for i in numbers]
    p = makePrime(10000000)                 
    perm = []                                      
    for i in range(1, nlen+1): perm += list(permutations(num, i))     
    num_arr = [int(("").join(p)) for p in perm]  
    for i in num_arr: 
        if p[i]: ans+=1; p[i] = 0
    return ans