23/11/24

프로그래머스 삼총사 

 

먼저 문제를 읽어보고 처음엔 3중 for문이 생각났지만 당연 먼가 안좋은듯한 느낌이 들었다. 그래서 파이썬에서 알아보니 

 itertools에 combinations인 조합을 이용 하면 된다!

서로 다른 n개가 있는데 순서를 따지지 않고 조합 할 수 있는 모든 경우를 iterator로 반환한다.

만약 [a,b,c] 있을때 list(itertools. combinations)하면 [ (a,b),(a,c),(b,c) ] 가 나온다

import itertools

def solution(number):
    answer = 0

    for i in itertools.combinations(number,3):
        if sum(i) == 0:
            answer +=1

    return answer

이렇게 combinations 이용하니 코드도 깔끔해 보인다.