[프로그래머스] 삼총사(Python)

문제 링크

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

프로그래머스

소스 코드

from itertools import combinations

def solution(number):
    nums = list(combinations(number,3))
    
    cnt = 0
    for num in nums:
        if sum(num) == 0:
            cnt += 1
    return cnt

해설

• combinations(조합) 함수를 이용한 간단한 풀이

 

combinations

combinations(list,number)

위와 같이 함수를 사용하면 list 내의 원소 중 number개를 뽑아 조합을 만들어준다.

예를 들어 list = [1, 2, 3], number = 2인 경우 (1, 2), (1, 3), (2, 3)이 반환된다.

 

따라서 문제에서 주어진 number라는 숫자 리스트에서 3개를 뽑아 조합을 만들어 리스트를 만들고,

각 튜플의 합이 0인 경우에 count를 해주면 정답이 된다.