[프로그래머스] lv.1 삼총사 python 풀이

문제

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

프로그래머스

 

문제 풀이

itertools의 combination 함수를 사용하여 가능한 모든 경우를 구하고, 단순 구현으로 풀이함.

 

코드

import itertools
def solution(number):
    op = lambda x: 1 if sum(x) == 0 else 0
    return sum([op(x) for x in itertools.combinations(number, 3)])