문제

N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

출력

첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.

풀이

이전에 풀었던 암호 만들기와 비슷한 방식이다.

백트래킹을 이용했고, 합쳐서 만들 수열(numbers)와 현재 인덱스(currentIndex)를 파라미터로 넣어준다.
지금까지 온 인덱스 이후값부터 number에 들어갈 수 있게 해주었다.

초반에 계속 실패했다고 뜨길래 왜지 했는데...합한 값이 S와 같게 되면 return을 해버려서 실패로 떴다. 만약 S=0, [-5, 0, 0, 2, 3]과 같은 수열이 주어졌을때 [0]에서 return 해버려서 [0,0]값까지 구하지 못해서 틀렸던듯 하다.

코드

var fs = require('fs');

const [NS, sequence] = fs
	.readFileSync('./input.txt')
	.toString()
	.trim()
	.split('\n');

const [N, S] = NS.split(' ');
const sequences = sequence.split(' ');

function solution() {
	let answer = 0;
	for (let i = 0; i < +N; i++) {
		BT([sequences[i]], i);
	}

	function BT(numbers, currentIndex) {
		sum = numbers.reduce((a, b) => +a + +b, 0);
		if (sum === +S) answer++;

		for (let j = currentIndex + 1; j < +N; j++) {
			number = sequences[j];
			BT([...numbers, number], j);
		}
	}

	console.log(answer);
}

solution();