[Python] 프로그래머스 lv1 - 삼총사

[Python] 프로그래머스 lv1 - 삼총사

 

코딩테스트 연습 - 삼총사 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

풀이방법

 

• 문제에서 원하는 답
  • number에서 3개를 골라 0이 되는 경우의 수
  • 이때 순서는 상관없음 - 조합
    • 조합 : 서로 다른 n개에서 순서와 상관 X, r개를 뽑아야 함
    • 순열 : 서로 다른 n개에서 순서와 상관 O, r개를 뽑아야 함
• 의사 코드
  • 3개를 고르는 조합을 생성
  • 생성된 조합의 리스트를 사용하여 for문
  • for문을 반복하면서 만약 해당 원소의 합이 0이면 정답 +=1 
• test faild의 경우
  
  • 순열을 뽑고 set으로 겹치는 원소를 제거하고자 하였지만 안에서 원소 정렬의 실패
    • 방향을 조합으로 변경하여 해결

 

풀이코드

import itertools

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