[프로그래머스]소수찾기/42839번/파이썬/Python/완전탐색

heeee·2021년 2월 22일
0

algorithm

목록 보기
80/123

💡문제

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

제한 사항

  • numbers는 길이 1 이상 7 이하인 문자열입니다.
  • numbers는 0~9까지 숫자만으로 이루어져 있습니다.
  • 013은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

입출력

numbersreturn
173
0112


📖내가 작성한 code

from itertools import permutations

def check(x):
    if x<2:
        return False
    for i in range(2, x):
        if x % i == 0:
            return False
    return True

def solution(numbers):
    # 집합에 permuation 통해 탐색하며 값 추가
    answer=[]
    per=[]
    for i in range(1,len(numbers)+1):
        per=list(map(''.join,permutations(numbers,i)))
        per=set(per)
        for p in per:
            if check(int(p)):
                answer.append(int(p))
    answer=set(answer)
    return len(answer)

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42839

0개의 댓글