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

/  제출 1  /

from itertools import combinations
def isPrimeNum(x):
    for i in range(2,x):
        if x % i == 0:
            return False
    return True
def solution(nums):
    answer = 0
    # 3개 골라서
    for candidate in list(combinations(nums,3)):
        if isPrimeNum(sum(candidate)):
            answer += 1
    

    return answer

채점 결과

정확성: 100.0

합계: 100.0 / 100.0

 

순간 소수 판별이 기억이 안 나서 흠칫 했습니다. 

다행히 금방 떠올렸고. 혹시나 헷갈릴까봐 따로 함수로 빼서 구현했습니다.

combinations도 이름이 기억이 안 나서 permutations를 이용했다가 오답이 나와 검색 후 코드 작성했습니다.

자주 쓰이는 메서드나 라이브러리는 꼭 헷갈리고 기억해야겠습니다.

 

---