프로그래머스 [Summer/Winter Coding ~2018] 소수 만들기 - Henry's Algorithm

안녕하세요

행복한 개발자 Henry입니다!

 

오늘은 프로그래머스에서 

오로지 코딩실력만으로 여름 인턴십을 선발하는 Summer/Winter Coding 챌린지 중

2018년도에 출제되었던 문제를 풀어보았습니다.

 

이번 문제는 프로그래머스 Level 1에 배정되었던 문제였습니다.

 

소수를 구하는 문제였는데요,

for문 3번을 모두 돌려야 하는 완전 탐색이 불가피했던 문제였고,

그렇게 접근해서 풀었더니 모든 테스트에서 정답을 받을 수 있었습니다.

 

아래는 제가 풀었던 문제의 코드입니다.

저는 따로 is_prime이라는 함수를 만들어 문제를 풀어보았습니다.

def solution(nums):
    prime_cnt = 0

    for first in range(len(nums)):
        for second in range(first+1,len(nums)):
            for third in range(second+1, len(nums)):
                if is_prime(nums[first] + nums[second] + nums[third]):
                    prime_cnt += 1

    return prime_cnt

def is_prime(num: int) -> bool:
    for i in range(2, num//2):
        if num%i == 0:
            return False
    return True

 

그럼 오늘도 좋은 하루 되세요~~!!