반응형
from itertools import combinations
def check(a,b,c):
sum = a+b+c
for i in range(2,sum):
if sum%i == 0 : return False
return True
def solution(nums):
answer = 0
A = list(combinations(nums,3))
for i in A :
if check(i[0],i[1],i[2]) : answer += 1
return answer
"check"라는 소수인지 확인해주는 함수를 추가해줬다.
여기서도 combinations을 활용할 수 있었다.
한번 기억하니 아주 유용하다ㅎㅎㅎ
반응형
'알고리즘 > 프로그래머스 level 1,2' 카테고리의 다른 글
[파이썬🐍] 프로그래머스 : 나누어 떨어지는 숫자 배열 (0) | 2021.04.20 |
---|---|
[파이썬🐍] 프로그래머스 : 예산 (0) | 2021.04.20 |
[파이썬🐍] 프로그래머스 : 음양 더하기 (0) | 2021.04.20 |
[파이썬🐍] 프로그래머스 : 내적 (0) | 2021.04.20 |
[파이썬🐍] 프로그래머스 : 3진법 뒤집기 (0) | 2021.04.20 |
댓글