프로그래머스 완전탐색 LV2

고장난 고양이·2022년 1월 16일
0

알고리즘_python

목록 보기
15/84

프로그래머스 완전탐색 LV2

소수 찾기

문제 설명
한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.
각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

제한사항
numbers는 길이 1 이상 7 이하인 문자열입니다.
numbers는 0~9까지 숫자만으로 이루어져 있습니다.
"013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.
입출력 예
numbers return
"17" 3
"011" 2

입출력 예 설명
예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.
예제 #2
[0, 1, 1]으로는 소수 [11, 101]를 만들 수 있습니다.
11과 011은 같은 숫자로 취급합니다.

코드

import itertools
def primenumber(x):
    for i in range(2, x):
    	if x % i == 0:	
        	return False
    return True	
def exist(list ,tmp):
    for i in range(len(list)):
        if list[i] == tmp:
            return True
    return False

def solution(numbers):
    num=[]
    answer=[]
    for i in range(len(numbers)):
        num.append(numbers[i])
    
    for i in range(1,len(numbers)+1):
        result = list(itertools.permutations(num,i))
        
        for j in range(len(result)):
            tmp=int(("").join(result[j]))
            print(tmp)
            if tmp == 0 : continue
            if tmp == 1 : continue
            if exist(answer,tmp): continue
            if primenumber(tmp) : answer.append(tmp)
        
    return len(answer)

무지성 코드 소수인지 판단하는 primenumber함수와 중복방지를 위한 exist함수를 생성 먼저 받은 문자열을 하나씩 뜯어서 리스트로 만들고
itertools의 permutations(순열)생성을 통해서 경우의 수들을 만든다.그리고 각경우의 수들을 합쳐서 이미존재하는지 0,1을 제외한경우를 리스트에 넣는다.

참고

from itertools import permutations
def solution(n):
    a = set()
    for i in range(len(n)):
        a |= set(map(int, map("".join, permutations(list(n), i + 1))))
    a -= set(range(0, 2))
    for i in range(2, int(max(a) ** 0.5) + 1):
        a -= set(range(i * 2, max(a) + 1, i))
    return len(a)
profile
개발새발X발일지

0개의 댓글