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

문제 출처:

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

 

코드 리뷰:

내장함수 combinations 을 사용하였다.

3가지 조합의 모든 경우의 수를

num_com 리스트 변수에 저장 후에

for문을 돌려서

원소 합=0 을 만족하는 리스트만 count 개수를 셌다.

 

작성 코드 

from itertools import combinations

def solution(number):
    num_com = list(combinations(number,3))

    cnt=0 # 삼총사 집계를 위한 변수 
    for array in num_com:
        if sum(array)==0:
            cnt+=1

    return cnt