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

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

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

작성한 질문수

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

11. 수들의 조합(DFS)

이렇게 풀어도 되나요?

작성

·

82

0

n,k = map(int, input().split())
data = list(map(int, input().split()))
m = int(input())

cnt = 0
def dfs(l,s,c):
    global cnt
    if l == n:
        if c == k:
            if s % m == 0:
                cnt += 1
        return
    dfs(l+1,s+data[l],c+1)
    dfs(l + 1, s, c)
dfs(0,0,0)
print(cnt)

이렇게 풀어도 되나요?

답변 1

0

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

네, 제시해주신 파이썬 코드는 주어진 문제 조건(수열에서 k개의 수를 선택해서 그 합이 m으로 나누어떨어지는 경우의 수를 찾는 문제)에 대해 잘 작성되었고, DFS(깊이 우선 탐색)를 이용하여 문제를 해결하는 유효한 방법입니다.

제시된 코드는 다음과 같은 프로세스로 작동합니다:
1. dfs 함수는 깊이(l), 현재까지의 합(s), 선택된 수의 개수(c)를 인자로 받습니다.
2. 탐색의 깊이가 입력 값 n에 도달하면, 선택된 수의 개수가 주어진 k와 같은지, 그리고 현재까지의 합이 m으로 나누어떨어지는지 확인하여 조건에 맞는 경우 cnt를 증가시킵니다.
3. 각 단계에서 다음 수를 선택하는 경우와 선택하지 않는 경우로 나누어 다음 단계의 dfs 함수를 호출합니다.

코드의 접근 방식과 구현이 문제의 요구 사항을 충족하므로 이대로 사용하셔도 됩니다. 다만, 실제 코딩 테스트 환경에서는 실행 시간, 메모리 사용량 등을 고려하여 최적화가 필요할 수 있습니다. 추가적으로 다른 풀이 방식이나 최적화 방안에 대해서도 고민해 보시는 것을 추천드립니다.

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

작성한 질문수

질문하기