[프로그래머스_완전탐색] 소수 찾기(파이썬)

문제 풀기 전 알아야 될 지식

- 순열(permutations)

서로 다른 n개에서 r개를 선택할때 순서를 고려하여 중복없이 뽑을 경우의 수.

 

 

 

파이썬에는 순열(permutations)와 조합 을 손쉽게 만들어주는 모듈 iteratools가 있다.

그래서 

 

from iteratools import permutations

위 코드를 이용하여 모듈을 가져오면 된다.

 

 

예시 코드

from iteratools import permutations # 순열과 조합을 손쉽게 만들어주는 모듈 itertools 사용
arr = ['a','b','c','d'] # 예시로 배열 설정하여 arr 변수 저장
print(permutations(arr,2)) # 반복가능한 객체(리스트,튜플,문자열)안에서 r개를 선택한다. 여기서 r은 2로 잡았음 

for i in permutations(arr,2): # 배열 에서 2개 선택하여 i에 넣어서 반복시킨다.
	print(i) # i 값 출력

풀이

from itertools import permutations # itertools 모듈에서 permutations 가져오기

array = [] # 변수 array 명 [] 배열 생성

def check_prime_number(numberA): # check_prime_number 함수 , 인수 numberA 로 받음.
    number = int(numberA) # 변수 number 에 정수형 으로 numberA 저장하기
    if number != 1 and number != 0: # 만약 받은 정수형 numberA인 number변수가 1이 아닐경우 와 0이 아닐경우                
        for f in range(2, number): # 2부터 number까지 반복 for문 돌림  
            if number % f == 0:     # number에서 f로 나눈 나머지 값이 0일경우 
                return False # false 반환하게됨.
    else: # number가 1이거나 0일경우 (즉 아닐경우)
        return False # false 반환함

    array.append(number) #변수array(리스트)에 number 값 추가해주기.


    
def solution(numbers): #함수 solution에 numbers 매개변수로 받기.
    
    for length in range(1, len(numbers)+1):
        perList = list(map(''.join, permutations(list(numbers), length))) 
        
        for per in perList:
            check_prime_number(per)
                

    result = set(array)
    return len(result)

1. 우선 소수가 맞다면 배열(array)에 넣어주는 소수 Check 함수를 구현해주고
2. 입력 배열의 개수만큼 반복문을 통하여, 
3. itertools의 permutations을 이용하여 각 숫자로 이루어질 수 있는 숫자들을 구하여 배열(perList)에 포함시켜주고
4. 배열(perList)을 반복문을 통해, 위 1번에서 구현한 소수 Check로 Filtering 하여 배열(array)에 포함해주며
5. 마지막으로 Filtering된 배열(array)을 Set으로 치환하여 중복을 제거한 후 갯수를 출력하였습니다.