daniel7481의 개발일지

[프로그래머스]소수 찾기 본문

프로그래머스

[프로그래머스]소수 찾기

daniel7481 2022. 7. 31. 13:17
반응형

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

 

프로그래머스

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

programmers.co.kr

풀이

Permutations 함수를 사용하여 가능한 모든 숫자의 경우의 수를 가져온 다음, 딕셔너리  자료구조를 이용하여 해싱을 통해 이미 나온 적 있는 숫자인지 아닌지를 확인해서, 만약 나온 적이 없는 함수이면 직접 구현한 소수인지 판단하는 함수인 prime함수를 통해 만약 참이면 answer에 하나씩 더해주게 하였다.

from itertools import permutations
def prime(x):
    if x== 1 or x == 0:
        return False
    for i in range(2, x):
        if x % i == 0:
            return False
    return True
def solution(numbers):
    answer = 0
    dic = {}
    numbers = list(numbers)
    com = []
    for i in range(1, len(numbers)+1):
        com += list(permutations(numbers, i))
    for co in com:
        num = int("".join(co))
        try:
            dic[num]
        except:
            dic[num] = True
            #print(num)
            if prime(num):
                answer += 1
    return answer
반응형