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

js1881님의 프로필 이미지
js1881

작성한 질문수

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

1. 최대점수 구하기(DFS)

최대점수 구하기(DFS) 질문 드립니다.

작성

·

331

·

수정됨

0

안녕하세요, 해당 문제를 풀다가 질문 사항이 있어서 질문드립니다.

import sys
sys.stdin = open("C:\\Study\\algorithm\\input.txt", "r")

def DFS(L, sum, time):
    global res
    if res[0] <= time <= m:
        if res[1] < sum:
            res[0] = time
            res[1] = sum
    for i in range(1, n+1):
        if time > m:
            break
        if ch[i] == 0:
            ch[i] = 1
            DFS(L+1, sum+ls[i-1][0], time+ls[i-1][1])
            ch[i] = 0


if __name__ == "__main__":
    n, m = map(int, input().split())
    ls = []
    ch = [0] * (n+1)
    res = [0] * 2
    for i in range(n):
        a, b = map(int, input().split())
        ls.append([a, b])
    DFS(0, 0, 0)
    print(res[1])

위 코드를 돌려보면 테스트 케이스 1,2번은 풀어지나 3번부터 timeOut이 뜨네요 (무한 루프가 도는거 같습니다)

이전 문제에서 알려주신 비슷한 방식으로
해당 문제를 1~n까지 DFS(n)전체를 탐색해서 최적의 점수를 도출할려고 했는데, 해당 유형의 문제가 나오면 이러한방식으론 해결이 불가능한걸까요?

답변 1

1

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

안녕하세요^^

위에 코드는 순열과 같은 코드라 중복확인을 너무 많이 해서 그렇습니다.

즉 1번, 2번, 3번, 4번 문제를 풀 경우 부분집합을 경우는 한 번만 확인하는데 비해 위에 코드는 순열이라 순서가 다르면 다른 경우로 생각하는 코드입니다.

1번, 2번, 3번, 4번 순으로 푼 경우

1번, 2번, 4번, 3번 순으로 푼 경우

1번, 3번, 2번, 4번 순으로 푼 경우 등등 모두를 확인합니다.

즉 4문제를 푼 경우라고 하면{1, 2, 3, 4}라는 부분집합 한 번만 확인해야 할 것을 위에 코드는 4!의 경우를 다 확인하는 코드입니다.

 

js1881님의 프로필 이미지
js1881

작성한 질문수

질문하기