[Programmers] 삼총사(Python3, 파이썬)

 

본 게시글은 프로그래머스 코딩테스트 연습 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

레벨

Level 1

문제 출처

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

처음보자마자 이건 조합으로 풀어야겠구나 했는데... 평소에 itertools을 많이 사용하지 않아서 까먹어서 ... combinations를 검색을 해서 풀어버렸다ㅠㅠ 아무튼 combinations를 사용하면 바로 3개를 조합해주는데, 그를 list로 묶어 nums에 저장 후, 반복문을 돌리며 합이 0일 때만 카운트해주면 된다.

다음에 까먹지 않기 위해 정리해보는 combinations

from itertools import combinations

combinations(list,number)

이와 같이 작성하면 list내에 number개수만큼 수를 뽑아 조합을 만들어준다.
list = [0, 1, 2] 이면 number=2 라면, (0, 1) (0, 2), (1, 2)가 뽑아진다.

다른 사람의 풀이

위와 비슷하므로 생략