본문 바로가기

알고리즘 문제

프로그래머스 타겟 넘버

DFS를 활용했다.

 

answer는 target과 일치하면 1씩 누적하는 변수다. 두 함수에서 사용할 수 있도록 전역변수로 만들었다.

numbers와 길이, target, 0을 인자로 갖는 make_target을 호출한다.

i는 level을 측정하면서 numbers의 index로도 쓰인다. numbers의 끝까지 탐색을 했다면 numbers의 합이 target과 일치하는지 판단하고 일치한다면 answer에 1을 누적한다. 그리고 make_target 함수를 종료한다.

 

트리 레벨(i)에 따라 레벨에 있는 하나의 원소의 기호를 변경해준다.

numbers[i] *= 1
make_target(numbers, target, length, i+1)
numbers[i] *= -1
make_target(numbers, target, length, i+1)

1부터 시작
-1부터 시작

 

answer = 0


def make_target(numbers, target, length, i):
    global answer
    if i == length:
        if target == sum(numbers):
            answer += 1
        return
    numbers[i] *= 1
    make_target(numbers, target, length, i+1)
    numbers[i] *= -1
    make_target(numbers, target, length, i+1)


def solution(numbers, target):
    global answer
    length = len(numbers)
    make_target(numbers, target, length, 0)
    return answer

출처 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+1 = 3 +1+1-1+1+1 = 3 +1+1+1-1+1 = 3 +1+1+1+1-1 = 3 사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘

programmers.co.kr

 

'알고리즘 문제' 카테고리의 다른 글

프로그래머스 단어 변환  (0) 2019.09.03
프로그래머스 네트워크  (0) 2019.09.02
프로그래머스 입국심사  (0) 2019.09.01
프로그래머스 예산  (0) 2019.08.31
프로그래머스 카펫  (0) 2019.08.30