알고리즘

[프로그래머스] 삼총사

반신욕좋아하는J 2022. 11. 17. 10:00
반응형

https://school.programmers.co.kr/learn/courses/30/lessons/131705

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

나의 풀이 

def solution(number):
    result = 0
    num_len = len(number)
    for one in range(num_len-2):
        for two in range(one + 1, num_len-1):
            for t in range(two + 1, num_len):
                print(number[one], number[two], number[t])
                if number[one] + number[two] + number[t] == 0:
                    result += 1
    return result

순열 조합으로 푼 풀이 

def solution (number) :
    from itertools import combinations
    cnt = 0
    for i in combinations(number,3) :
        if sum(i) == 0 :
            cnt += 1
    return cnt
반응형