[프로그래머스] 소수 만들기

https://programmers.co.kr/learn/courses/30/lessons/12977#

코딩테스트 연습 - 소수 만들기

 

<코드>

from itertools import combinations

def solution(nums):
    cnt = 0
    for num in list(combinations(nums, 3)):
        N = sum(num)
        for i in range(2, int(N**0.5)+1):
            if N % i == 0:
                break
        else:
            cnt += 1
        
    
    
    return cnt

 

<풀이방법>

1. 소수는 1과 자신 빼고는 나누어지는 수가 없어야 하므로 for문으로 나머지가 0인 숫자가 존재하는지 찾으면 됨.

2. 범위를 자기의 제곱근까지로 해버리면 모든 숫자를 탐색하지 않고 반틈만 탐색하면 돼서 시간을 줄여준다.

 

<리뷰>

최근에 백준에서 풀었던 문제랑 거의 흡사해서 빨리 풀 수 있었다.