프로그래머스_타겟 넘버_python

2021. 8. 13. 20:34코딩테스트

반응형

 

https://programmers.co.kr/learn/courses/30/lessons/43165

 

코딩테스트 연습 - 타겟 넘버

n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다. -1+1+1+1+1 = 3 +1-1+1+1+

programmers.co.kr

 

import itertools
def solution(numbers, target):
    answer = 0
    if sum(numbers) == target:
        return 1
    tmp = (sum(numbers)-target)//2
    for i in range(1,len(numbers)):
        a = itertools.combinations(numbers, i)
        for b in a:
            if sum(b) == tmp:
                answer += 1
    return answer

[내코드]

총합에서 타겟만큼 빼서 -를 붙일 총합을 구함.

for문에서 조합을 뽑아 총합이 같다면 방법+1

 

def solution(numbers, target):
    if not numbers and target == 0 :
        return 1
    elif not numbers:
        return 0
    else:
        return solution(numbers[1:], target-numbers[0]) + solution(numbers[1:], target+numbers[0])

[다른사람 코드]

재귀 함수를 사용해 더이상 계산할 number가 없을때까지 재귀함수

반응형