(프로그래머스) lv.1 삼총사

 

https://school.programmers.co.kr/learn/courses/30/lessons/131705

프로그래머스

문제풀이

from itertools import combinations


def solution(number):
    answer = 0
    # 조합을 만들고자 하는 데이터, 조합의 크기를 입력하면 모든 조합을 튜플 리스트로 반환
    combo_list = list(combinations(number, 3))

    # 조합을 하나씩 꺼내서 조합의 합계가 0이라면 카운트
    for combo in combo_list:
        if not sum(combo):
            answer += 1

    return answer