알고리즘/코딩테스트 공부
8-15. 수들의 조합 (DFS)
YG - 96년생 , 강아지 있음, 개발자 희망
2022. 3. 25. 09:36
문제
N개의 정수가 주어지면 그 숫자들 중 K개를 뽑는 조합의 합이 임의의 정수 M의 배수인 개수 는 몇 개가 있는지 출력하는 프로그램을 작성하세요. 예를 들면 5개의 숫자 2 4 5 8 12가 주어지고, 3개를 뽑은 조합의 합이 6의 배수인 조합을 찾으면 4+8+12 2+4+12로 2가지가 있습니다.
풀이
function solution(n, k, arr, m) {
let answer = 0;
let tmp = Array.from({ length: k }, () => 0);
function DFS(L, s, sum) {
if (L === k) {
console.log(tmp);
if (sum % m === 0) {
answer++;
}
} else {
for (let i = s; i < n; i++) {
tmp[L] = arr[i];
DFS(L + 1, i + 1, sum + arr[i]);
}
}
}
DFS(0, 0, 0);
return answer;
}
let arr = [2, 4, 5, 8, 12];
console.log(solution(5, 3, arr, 6));