[Algorithm] 소수 찾기

코딩테스트 연습 - 소수 찾기

 

해결 과정

 

 

나의 풀이

import math
import itertools

# 소수 판별 함수
def isPrimeNumber(number):
    if number == 1 or number == 0:
        return False
    for i in range(2, int(math.sqrt(number)) + 1):  # n의 제곱근을 정수화 시켜준 후 + 1
        if number % i == 0:
            return False
    return True

# numbers로 만들 수 있는 모든 수 배열로 return 하는 함수
def strToAllNumber(numbers):
    numList = []

    for i in range(1, len(numbers) + 1):  # len == 2이면, 1, 2
        result = list(itertools.permutations(
            numbers, i))  # [('1', '3'), ('1', '4'), ('2', '1')]
        result = map(lambda x: int(''.join(x)), result)
        numList += result
        numList = list(set(numList))  # 중복 제거

    return numList

def solution(numbers):
    numList = strToAllNumber(numbers)
    count = 0
    for value in numList:
        if isPrimeNumber(value):
            count += 1

    return count

 

다른 풀이

 

 

IDEA