삼총사-파이썬(Python) [프로그래머스/코딩테스트 연습/Lv.1]

문제

https://school.programmers.co.kr/learn/courses/30/lessons/131705?language=python3

프로그래머스

 

---

풀이

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

---

Tip

이전에 풀었던 문제인 '두 개 뽑아서 더하기' 문제와 상당히 비슷하다. (참고: https://20mini.tistory.com/47) 그때 사용했던 combinations 함수를 사용하면 다음과 같이 풀 수 있다.

(참고: https://20mini.tistory.com/47)

from itertools import combinations

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

---

TIL

없음