<aside> 💡

라이브러리만 잘 활용할 줄 안다면 5분만에 풀 수 있는 문제

</aside>

문제

131705?language=python3

아이디어

첫시도

• 중첩 for문을 돌린다
  • for c in number: 로 첫 반복문을 돌고 중첩 for 문을 돌아 a , b 의 합을 구한 후 c 와 비교하여 제시하는 아이디어를 골랐다. 하지만 반복문을 3번 도는 과정에서 뭔가 꼬이는 사태가 발생하여 시간안에 풀지 못했다..
  • 반복문을 3개 돌리는 게 맞나..? 싶었고 과연 이게 이렇게 밖에 풀 수 없을까? 하고 생각이 들었다. 물론 시간복잡도도 많이 잡아먹게 될 거 같음

그래서 코드를 참고 하였다.

코드

from itertools import combinations

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

• itertools의 combination을 사용하는 법

• 아래 방법을 사용하여 3개의 조합을 모두 뽑은 다음 그 조합의 합이 0이면 answer 증가하는 방식으로 구현

  from itertools import combinations
  
  result = list(product(data, repeat=2)) 
  #전체 리스트 중에서 2개뽑는 모든 조합을 구하는 것
  

보완할 점

• 라이브러리의 많은 툴을 알고 활용하자….