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

코딩코딩코딩코님의 프로필 이미지
코딩코딩코딩코

작성한 질문수

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

4. 합이 같은 부분집합(DFS)

이방식은 시간초과로 되지않을까요?

작성

·

59

0

이방식은 예시는 잘 돌아가는데 추후 시간초과같은걸로 컷 엣지를 하지못해 틀린 답일까요?

n = int(input())
data = list(map(int, input().split()))
ch = [0]*n
def dfs(num):
    if num == n:
        choose = 0
        unchoose = 0
        for i in range(n):
            if ch[i] == 1:
                choose += data[i]
            else:
                unchoose += data[i]

        if choose == unchoose:
            print("YES")
            exit()
        return

    ch[num] = 1
    dfs(num+1)
    ch[num] = 0
    dfs(num+1)

dfs(0)
print("NO")

답변 1

0

안녕하세요, 인프런 AI 인턴입니다.

코드에 대한 질문 감사합니다. 제시하신 방식은 모든 부분 집합을 확인하는 방식으로, 입력 크기(n)에 따른 시간 복잡도가 지수 시간인 ( O(2^n) )임을 알 수 있습니다. 따라서 n의 크기가 조금만 커져도 매우 많은 연산을 수행해야 하므로, 큰 데이터 셋을 대상으로 할 때 시간 초과가 발생할 가능성이 매우 높습니다.

이런 부분 집합의 합 문제는 분할 정복, 백트래킹, 동적 계획법(DP) 등 다양한 방법으로 접근할 수 있습니다. 그 중에서 시간 복잡도를 줄이기 위해선, 중복 계산을 줄이는 방향으로 알고리즘을 최적화하는 것이 필수입니다. DP 또는 메모이제이션 기법을 적용해 보는 것도 좋을 수 있으며, 문제의 조건과 특성에 따라 부분 문제로 나누어 해결하는 방법도 고려해 볼 수 있습니다.

특히 코딩 테스트 대비를 위해서는 다양한 문제를 접하며 여러 해결 방법을 시도해 보고, 시간 복잡도를 줄일 수 있는 방법을 항상 고려해야 합니다. 풀이 방식에 따라 시간 초과 없이 문제를 해결할 수 있는 효율적인 방법을 찾아보시기 바랍니다.

코딩코딩코딩코님의 프로필 이미지
코딩코딩코딩코

작성한 질문수

질문하기