[프로그래머스/Programmers] 타겟 넘버 - Python

Link

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

 

Solution

해당 문제는 재귀함수를 통해 해결할 수 있다.

len(numbers) == 0 and target == 0 은 연산의 결과와 target이 같음을 나타내고

len(numbers) == 0 and target != 0 은 연산의 결과와 target이 다름을 나타낸다.

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

🔑 key point: Recursive Function

 

Other Solution

해당 풀이는 product()를 사용하여 해결하였다. 풀이 하단에 product() 사용예시를 첨부할테니 참고바란다.

from itertools import product

def solution(numbers, target):
    numbers = [(x, -x) for x in numbers]

    sums = list(map(sum, product(*numbers)))

    return sums.count(target)

🔑 key point: product(*numbers)

 

•  product()

from itertools import product

a = [1, 2, 3]

a = [(x, -x) for x in a]

a = list(product(*a))

print(a)

<output>
[(1, 2, 3), (1, 2, -3), (1, -2, 3), (1, -2, -3), (-1, 2, 3), (-1, 2, -3), (-1, -2, 3), (-1, -2, -3)]

from itertools import product

a = [1, 2, 3]

b = [4, 5, 6]

c = list(product(a, b))

print(c)

<output>
[(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]