[Python] 프로그래머스 코딩테스트 연습 level1 (소수만들기)

1. 문제 설명

 

 

 

2. 제한사항

 

 

 

3. 내 풀이

from itertools import combinations
import math

def primenumber(x):
    for i in range (2, int(math.sqrt(x)) + 1):	
    	if x % i == 0:		
        	return False
    return True		
                    
def solution(nums):
    answer = 0

    data = combinations(nums,3)
    for i in data:
        x = sum(i)
        if primenumber(x):
            answer += 1

    return answer

 

1. 입력받은 인자가 소수이면 True를 반환하는 primenumber 함수 생성

2. combinations 모듈을 이용하여 입력받은 리스트에서 3개 숫자씩 조합

3. 조합한 숫자를 sum해서 소수이면 answer +1

 

 

 

 

4. 다른 사람 풀이

def solution(nums):
    from itertools import combinations as cb
    answer = 0
    for a in cb(nums, 3):
        cand = sum(a)
        for j in range(2, cand):
            if cand%j==0:
                break
        else:
            answer += 1
    return answer