본문 바로가기
스터디IT🌼/Algorithm

[ 프로그래머스 ] 소수찾기 _ 완전탐색

by 동백사과 2022. 10. 5.

🌱 문제 분류

- 완전탐색 

 

🌱 문제 링크

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

 

프로그래머스

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

programmers.co.kr

 

🌱 문제 설명

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

각 종이 조각에 적힌 숫자가 적힌 문자열 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은 같은 숫자로 취급합니다.

 

🌱문제풀이

-  문자열을 쪼개서 만들 수 있는 경우의 수를 모두 구한다 ( 순열 )

-  나올 수 있는 경우의 수를 모두 정수로 바꾸고 중복을 제거한다.

   -> 같은 수가 여러개 일 수 있기 때문 ( 11, 011이 같은 걸로 보기 위해 int 형으로 바꿔서 set으로 중복 제거 )

-  나온 경우의 수 중 소수 판별  : 2~ 해당 수의 제곱근까지 의 수로 해당 수를 나눴을 때 나눠지면 소수가 아님! 

   -> 에라토스테네스의 체! 원리

 

🌱 [ 문제 풀이 코드 ]

import itertools # 순열 사용을 위해
def solution(numbers):
    nums = [ n for n in numbers]
    
    permutations = []
    
    for i in range(1, len(numbers)+1):
        permutations += list(itertools.permutations(numbers,i)) # 순열
    all_num = [int(("").join(p)) for p in permutations] # 정수형으로 
    all_num = set(all_num) # 중복 제거
    answer=[]
    for num in all_num:
        if num < 2:
            continue
        check= True
        for i in range(2, int(num**0.5)+1): # 소수 판별 
            if num % i == 0:
                check = False
                break
        if check:
            answer.append(num)
        
    return len(answer)

 

# 기타 참고 코드
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)) # 소수가 아닌 0~1까지 제거 
    for i in range(2, int(max(a) ** 0.5) + 1):
        a -= set(range(i * 2, max(a) + 1, i)) # 범위 내의 배수 제거 
    return len(a)

 

댓글