아이공의 AI 공부 도전기

[프로그래머스]  Level 2 : 타겟 넘버 (Python)

 

     

 

 

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

 

프로그래머스

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

programmers.co.kr

코드 링크

https://github.com/stellaluminary/Programmers

 

GitHub - stellaluminary/Programmers

Contribute to stellaluminary/Programmers development by creating an account on GitHub.

github.com

Python

 

방법 1 

 

def dfs(visit, idx, numbers, res, target, cnt):
    if idx == len(numbers):
        if res == target:
            cnt += 1
        return cnt

    visit[idx] = 1

    res += numbers[idx]
    cnt = dfs(visit, idx + 1, numbers, res, target, cnt)
    res -= numbers[idx]

    res -= numbers[idx]
    cnt = dfs(visit, idx + 1, numbers, res, target, cnt)
    res += numbers[idx]

    visit[idx] = 0

    return cnt


def solution(numbers, target):
    answer = 0
    visit = [0] * len(numbers)
    answer = dfs(visit, 0, numbers, 0, target, 0)

    return answer

 

 

 

 

방법 2 

타 코드

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])

 

방법 3 

타 코드

from itertools import product

def solution(numbers, target):
    l = [(x, -x) for x in numbers]
    s = list(map(sum, product(*l)))
    return s.count(target)

 

방법 4

 

def dfs(idx, numbers, res, target):
    global answer

    if idx == len(numbers):
        if res == target:
            answer += 1
        return

    dfs(idx + 1, numbers, res + numbers[idx], target)
    dfs(idx + 1, numbers, res - numbers[idx], target)

def solution(numbers, target):
    global answer
    answer = 0
    dfs(0, numbers, 0, target)
    return answer

 

공유하기

facebook twitter kakaoTalk kakaostory naver band
loading