인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

수진안님의 프로필 이미지
수진안

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

2. 휴가(DFS)

이 코드는 왜 안되는지 설명좀 부탁드려도 될까요?

작성

·

198

0

import sys

sys.stdin=open("input.txt", "rt")

def DFS(L,M,J,sum):

    global cnt

    

    if L<n[0] and M<n[1] and J<n[2]:

        if sum>t:

            return

        elif sum==t:

            cnt+=1

            return

        else:

            DFS(L+1,M,J,sum+p[0])

            DFS(L,M+1,J,sum+p[1])

            DFS(L,M,J+1,sum+p[2])

    

if __name__=="__main__":

    t=int(input())

    k=int(input())

    p=[]

    n=[]

    cnt=0

    for i in range(k):

        a,b=map(int,input().split())

        p.append(a)

        n.append(b)

    DFS(0,0,0,0)

    print(cnt) 

        

    

답변 3

0

김태원님의 프로필 이미지
김태원
지식공유자

hank y님 감사합니다. ^^ 

제가 답변이 늦어질때 이렇게 공부하시는 분들끼로 서로 도와주며 했으면 좋겠네요.

0

김태원님의 프로필 이미지
김태원
지식공유자

위 코드는 동전의 종류가 3개일때만 가능한 코드입니다.  접근하는 방향를 수정해야 합니다.

그리고 L값에 대해서도

L=0일때 1개적용

L=1일때 2개적용

L=2일때 3개적용 하고 넘어가면 L은 3이 되어 있습니다. 즉  if L<n[0] and M<n[1] and J<n[2]: 이 상황에서 L=3일때 3개가 적용된 상황입니다. if L<=n[0] and M<=n[1] and J<=n[2]: 해주어야 합니다. 

그리고 이 코드는 중복해서 카운팅하는 것을 제거하기가 힘든 코드입니다. 영상에서 제가 설명한 방법으로 했으면 좋겠습니다. 

아래 코드를 실행해보시면 위 말들이 느낌이 올겁니다.

import sys
sys.stdin=open("input.txt", "rt")
def DFS(L,M,J,sum):
    global cnt
    if L<=n[0] and M<=n[1] and J<=n[2]:
        if sum>t:
            return
        elif sum==t:
            cnt+=1
            print(L, M, J, sum)
            return
        else:
            DFS(L+1,M,J,sum+p[0])
            DFS(L,M+1,J,sum+p[1])
            DFS(L,M,J+1,sum+p[2])
            
if __name__=="__main__":
    t=int(input())
    k=int(input())
    p=[]
    n=[]
    cnt=0
    for i in range(k):
        a,b=map(int,input().split())
        p.append(a)
        n.append(b)
    DFS(0,0,0,0)
    print(cnt) 

0

자세한건 선생님이 답변해주시겠지만 일단 k라는 변수가 중간에 들어가서 꼬인것 같아요. k의 용도가 없는데 input() 한줄을 잡아먹고 있어요. 

수진안님의 프로필 이미지
수진안

작성한 질문수

질문하기