[프로그래머스]삼총사(python)

(Lv.01) 삼총사👾
https://school.programmers.co.kr/learn/courses/30/lessons/131705

3개의 요소를 뽑는 모든 조합을 확인해야 하므로, itertools 모듈의 combinations 함수를 사용하려고 한다😊

💻 답안

from itertools import combinations

def solution(phoneBook):
    answer=0
    list_number= combinations(phoneBook, 3) #겹치는 것 없이 3명의 그룹을 만들기
    print(list_number)
    for l in list_number:
        if sum(list(l))==0: #그룹의 합이 0인 경우 
            answer+=1
    return answer