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

안녕하세요

기존의 조합처럼 1부터 n까지의 조합결과를 구한다음에

그 값을 a 리스트의 인덱스로 사용하여 sum값을 구해도될까요?

테스트는 통과했습니다

def DFS(L, s):
    global tmp, cnt
    if L == k:
        tmp = 0
        for i in range(L):
            tmp += a[res[i]-1]
        if tmp % m == 0:
            cnt += 1
    else:
        for i in range(s, n+1):
            res[L] = i
            DFS(L+1, i+1)

if __name__ == "__main__":
    n, k = map(int, input().split())
    a = list(map(int, input().split()))
    m = int(input())
    tmp = 0
    cnt = 0
    res = [0] * (k+1)
    DFS(0, 1)
    print(cnt)