[4코1파] 4명의 안드로이드 개발자와 1명의 파이썬 개발자의 코딩 테스트 서막 : 4코1파

Rule :

하루에 1문제씩 풀기.
한 문제당 30분씩은 고민하기.
왜 그렇게 풀었는지 공유하기.
하루라도 놓친다면 벌금은 1,000원

START :

[3코1파] 2023.01.04~ (56일차)
[4코1파] 2023.01.13~ (47일차)

Today :

2023.02.28 [56일차]

프로그래머스 LV2
타겟 넘버
https://school.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이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항

주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
각 숫자는 1 이상 50 이하인 자연수입니다.
타겟 넘버는 1 이상 1000 이하인 자연수입니다.

입출력 예

입출력 예 설명

입출력 예 #1
문제 예시와 같습니다.

입출력 예 #2
+4+1-2+1 = 4
+4-1+2-1 = 4
총 2가지 방법이 있으므로, 2를 return 합니다.

문제 풀이 방법

DFS/BFS 가 가물가물 해지고 있어서
일부러 DFS/BFS 에서 선택한 문제로서,
예 그렇습니다.

풀이방법은
그래프 자료구조를 생각하면서 갈라지는 경우의 수를 생각한다!
주어진 예시에서 numbers가 [1,1,1,1,1] 5개 이면 2^5 = 32개
[4,1,2,1] 4 개이면 2^4개의 계산식의 경우의수가 나오는데
이 계산식을 number를 queue에 넣고, 인덱스 값과 같이 넣으면서(중요)
pop 하고 다음 index 에 있는걸 더하고 빼고하는 일련의 연산을 다시 queue에 넣고 반복하고
길이만큼 이동했으면(경우의 수만큼 모든 합을 더했으면), 하나씩 pop해서 target과 맞으면 answer를 올려주는 일련의 로직이 되겠다.

증빙

내 코드

def solution(numbers, target):
    answer = 0
    queue = [[numbers[0],0], [-1*numbers[0],0]]
    n = len(numbers)
    while queue:
        num, idx = queue.pop()
        idx+=1
        
        if idx<n:
            queue.append([num+numbers[idx], idx])
            queue.append([num-numbers[idx], idx])
            
        else:
            if num==target:
                answer+=1 
    
    return answer

다른 사람 풀이

에효 여기도 머리 압수당할 사람 많네

여담

재빠르게 클리어 다른 DFS,BFS 문제 조지러갑니다.
(일? 안합니다)

profile
꿈꾸는 것도 개발처럼 깊게

0개의 댓글