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

파이썬이랑 친해지기

파이썬에서는 순열과 조합을 위한 라이브러리를 제공해준다.

from itertools import combinations

a = [1, 2, 3, 4]

for i in combinations(a, 3):
    print(i)
    
# 결과
# (1, 2, 3)
# (1, 2, 4)
# (1, 3, 4)
# (2, 3, 4)

combinations 함수를 통해 특정 리스트에 있는 요소들을 조합으로 표현할 수 있다.

이때 반환되는 값은 iteratable 객체이며 for문으로 순회할 수 있다.

 

from itertools import permutations

a = [1, 2, 3, 4]

for i in permutations(a, 3):
    print(i)

# 결과
# (1, 2, 3)
# (1, 2, 4)
# (1, 3, 2)
# (1, 3, 4)
# (1, 4, 2)
# (1, 4, 3)
# (2, 1, 3)
# (2, 1, 4)
# (2, 3, 1)
# ...

permutations 함수를 이용하여 순열도 구할 수 있다.

코드

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