(프로그래머) 삼총사


내 코드

from itertools import combinations
def solution(number):
    cnt=0
    answer = list(combinations(number,3))
    for i in answer:
        if sum(i)==0:
            cnt+=1
    return cnt

숫자에서 세 가지 요소의 모든 조합 찾기
0이 되는 모든 조합에 대해 +1