[Programmers] L1. 삼총사 (Python)

---

[문제 링크] 👇

프로그래머스

---

풀이

💡combinations 함수 사용

 

combinations는 순서에 상관없이 주어진 길이만큼 원소를 뽑는 모든 조합을 생성해주는 함수이다. itertools 모듈에 포함되어 있어, 이 모듈을 먼저 임포트해야 한다.

• combinations는 순서를 고려하지 않고 특정 길이만큼 요소를 뽑는 모든 조합을 생성한다.
• 결과는 튜플 형태로 반환되며, list()로 변환하면 전체 조합을 리스트로 볼 수 있다.
• 조합의 길이를 지정해야 하며, itertools 모듈에서 가져와야 사용 가능하다.

 

기본 구문

from itertools import combinations

combinations(iterable, r)

• iterable : 조합을 생성할 대상(예: 리스트, 문자열 등).
• r : 조합의 길이 (뽑을 개수).

 

Solution

from itertools import combinations 

def solution(number):
    answer = 0
    for i in combinations(number, 3):
        if sum(i) == 0:
            answer += 1
    return answer

 

 

---