daniel7481의 개발일지

[프로그래머스]타겟 넘버 본문

프로그래머스

[프로그래머스]타겟 넘버

daniel7481 2022. 7. 21. 16:31
반응형

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

 

프로그래머스

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

programmers.co.kr

풀이

dfs/bfs문제였는데 이번에는 dfs로 풀어봤다. 그래프에는 bfs가 좋지만, 이런 문제에는 재귀가 더 쉬운 것 같다. 구현 자체는 어렵지 않았지만, 헷갈린 점이 매 재귀마다 for문을 돌려야할줄 알았다. 생각해보니 그냥 반복문 없이 인덱스만 하나씩 이동하면서 visited을 확인하고 백트래킹하면 가능한 경우의 수를 구할 수 있다. visited는 visited[len(numbers)][2]로 해주었고, 2는 더했는지/뻈는지 여부 확인용이다. 

def dfs(x, visited, i, number, target):
        global answer
        if i == len(number) -1:
            if not visited[i][0]:
                if x + number[i] == target:
                    answer += 1
            if not visited[i][1]:
                if x - number[i] == target:
                    answer += 1
            return
        if not visited[i][0]:
            visited[i][0] = True
            dfs(x+number[i], visited, i+1, number, target)
            visited[i][0] = False
        if not visited[i][1]:
            visited[i][1]= True
            dfs(x-number[i], visited, i+1, number, target)
            visited[i][1] = False
def solution(numbers, target):
    global answer
    answer = 0
    visited = [[False, False]for _ in range(len(numbers))] #더했는지, 뺐는지 방문 확인 여부
    dfs(0, visited, 0, numbers, target)
    return answer
반응형