프로그래머스 | 삼총사 (Python3)

 

오늘도 오랑이는 문제를 풉니다. 

삼총사

 

풀이 과정

리스트의 각 인덱스 값이 출력되도록 for문으로 i, j, k 의 범위를 설정해줍니다.

i는 0~, j 는 i를 제외하도록 i+1~, k는 i와 j 를 제외하도록 j+1 을 시작점으로 설정해줍니다. 

각 인덱스에 대해 출력되는 값을 더한 것이 숫자 3개의 조합이 됩니다. 

def solution(number):
    for i in range(len(number)):
        for j in range(i+1,len(number)):
            for k in range(j+1,len(number)):
                answer = number[i]+number[j]+number[k]

 

숫자 3개의 합이 0이 되면 '삼총사'이기 때문에 if 문을 사용합니다. 

횟수를 세려면 앞에 미리 answer = 0 변수를 설정해두고 거기에 하나씩 적재되도록 해야 한다는 점 잊지 맙시다. 

 

최종 코드

def solution(number):
    answer = 0
    for i in range(len(number)):
        for j in range(i+1,len(number)):
            for k in range(j+1,len(number)):
                if number[i]+number[j]+number[k] == 0:
                    answer += 1
    return answer

 

 

오랑우탄이 영어를 하고 오랑이가 코드마스터가 되는 그날까지~

 

 

프로그래머스