완전탐색 | 프로그래머스 고득점kit 소수 찾기 | Python


created_at : 15 May 2024 - 2 mins read time
Tags: 완전탐색

Table of Contents


프로그래머스 고득점kit 소수 찾기 | Python

📌 프로그래머스 고득점kit 소수 찾기 문제 바로가기


🔎 문제 설명

💚 Level 2

- 난이도 ★☆☆☆
- 완전탐색

itertools 의 permutations 를 이용하면 쉽게 풀리는 문제다. 시간초과도 나지 않는 무난한 문제!
아, 난 이제까지 set 에 원소를 더할 때 add 함수를 이용해서 더해줬는데 |= 로도 더할 수 있다고 한다! 아래는 이 방법을 이용해 원소를 더한 코드이다.



💻 내 코드

# 소수 찾는 함수
def find_prime(x):
    # 0과 1은 소수가 아니다
    if x < 2:
        return False
    for i in range(2,x):
        if x % i == 0:
            return False
    return True

from itertools import permutations
def solution(numbers):
    answer = 0
    numbers = list(numbers)
    possible = set()
    for i in range(1,len(numbers)+1):
        possible |= set(map(int, [''.join(a) for a in permutations(numbers, i)]))

    for p in possible:
        answer += find_prime(p)

    return answer




 

💙 You need to log in to GitHub to write comments. 💙
If you can't see comments, please refresh page(F5).