ツリーの組み合わせ


作成日:2022年2月7日午後5:02

インプリメンテーションコード

# 수들의 조합
import sys
#sys.stdin = open("input.txt" ,"rt")

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

if __name__ == "__main__":
    n, k = map(int, input().split())
    a = list(map(int, input().split()))
    m = int(input())
    cnt = 0
    DFS(0, 0, 0)
    print(cnt)