[프로그래머스] 삼총사

문제 링크:

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

회고:

itertools 라이브러리의 combinations 함수 이용 (조합)

python

from itertools import combinations


def solution(number: list):
    count = 0

    for p in combinations(number, 3):
        if sum(p) == 0:
            count += 1

    return count


if __name__ == "__main__":
    result = solution([-2, 3, 0, 2, -5])

    print(result)
    print(result == 2)

2
True