N개의 정수가 주어지면 그 숫자들 중 K개를 뽑는 조합의 합이 임의의 정수 M의 배수인 개수 는 몇 개가 있는지 출력하는 프로그램을 작성하세요.
예를 들면 5개의 숫자 2 4 5 8 12가 주어지고, 3개를 뽑은 조합의 합이 6의 배수인 조합을 찾으면 4+8+12 2+4+12로 2가지가 있습니다.
▣ 입력설명
첫줄에 정수의 개수 N(3<=N<=20)과 임의의 정수 K(2<=K<=N)가 주어지고, 두 번째 줄에는 N개의 정수가 주어진다.
세 번째 줄에 M이 주어집니다.
▣ 출력설명
총 가지수를 출력합니다.
5 3
2 4 5 8 12
6
2
[해설코드]
def DFS(L,sum,s):
global cnt
if L==m:
if sum%k==0:
cnt+=1
else:
for i in range(s,n):
DFS(L+1,sum+a[i-1],i+1)
if __name__=="__main__":
n,m=map(int,input().split())
a=list(map(int,input().split()))
k=int(input())
res=[0]*m
cnt=0
DFS(0,0,0)
print(cnt)
- DFS에 sum값과 함께 같은 수의 노드를 방문한 것을 기록하기 위해 s변수 사용
즉, 조합과 같은 것.
2.중단 조건은 sum이 k로 나누어 떨어질 떄 중단.