[Programmers] [Summer/Winter Coding(~2018)] [Python] Level1_소수 만들기

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

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

풀이

# 프로그래머스 1단계 - 소수 만들기
from itertools import combinations
def check(a, b, c):
    total = a + b + c
    for i in range(2, total):
        if total % i == 0: return False
    return True

def solution(nums):
    answer = 0
    A = list(combinations(nums, 3))
    for i in A:
        if check(i[i], i[1], i[2]): answer += 1
    return answer

itertools의 combinations을 사용하여 3가지를 선택하는 조합을 만들고

check 함수를 통해 소수인지 아닌지 판별

---

permutations / combination / product 사용법

items = ['1', '2', '3', '4', '5']
from itertools import permutations
list(permutations(items, 2))
# 모든 순열의 조합 (순서가 바뀐 중복 O)


from itertools import combinations
list(combinations(items, 2))
# 모든 순열의 조합(중복 x)

from itertools import product
items = [['a', 'b', 'c,'], ['1', '2', '3', '4'], ['!', '@', '#']]
# [('a','1','!), ('a','1','@') ...]