[programmers - python] Lv.1: 삼총사

문제 링크

프로그래머스

문제 풀이

def solution(number):
    answer = 0 
    l = len(number)
    for i in range(l):
        for j in range(i+1, l):
            for k in range(j+1, l):
                if number[i] + number[j] + number[k] == 0:
                    answer += 1
    return answer

삼총사 입출력 테스트

*key point: 중첩 반복문을 활용하여 number의 중복되지 않는 세 요소를 골라 그 합이 0인 경우만 세어준다.

 

*다른 사람의 풀이(knine3 , 최승연 , 강희민 , 최소영 , powep93 외 6 명)

itertools 모듈을 사용하면 combination을 이용하여 중첩 반복문을 하나의 반복문으로 줄일 수 있다.

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