IF - 수들의 조합

Goody·2021년 5월 23일
0

알고리즘

목록 보기
107/122

문제

N개의 정수가 주어지면 그 숫자들 중 K개를 뽑는 조합의 합이 임의의 정수 M의 배수인 개수 는 몇 개가 있는지 출력하는 프로그램을 작성하세요.
예를 들면 5개의 숫자 2 4 5 8 12가 주어지고, 3개를 뽑은 조합의 합이 6의 배수인 조합을 찾으면 4+8+12 2+4+12로 2가지가 있습니다.

예시

numbersKMresult
[2,4,5,8,12]362

풀이 및 회고

  • 주어진 배열 numbers 에서 k개를 뽑는 모든 경우의 수를 DFS로 구한다.
  • 조합이 모이는 temp 배열을 복사하고, 복사한 배열 내 원소들의 총 합이 주어진 수 m으로 나누어 떨어지면 정답의 개수를 하나 늘린다.
  • 조합을 DFS로 구하는 방법만 알면 쉽게 풀수 있는 문제였다.

코드

const solution = (numbers, k, m) => {
	const combinations = [];
	const temp = [];
	const n = numbers.length;
	let answer = 0;

	const DFS = (L, s) => {
		if (L === k) {
			const copy = [...temp];
			const sum = copy.reduce((acc, cur) => acc + cur);
			if (sum % m === 0) answer++;
		} else {
			for (let i = s; i < n; i++) {
				temp[L] = numbers[i];
				DFS(L + 1, i + 1);
			}
		}
	};

	DFS(0, 0);
	console.log(answer);
};

0개의 댓글