Q 1182 부분수열의 합
💡 문제 요약 및 분석
N 개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하자.
💡 알고리즘 설계
✅ 백트래킹 조건 : 현재 요소를 선택하거나, 선택하지 않거나 두 개의 선택지로 나뉜다.
입력
- 표준입력 : int N, int S
- 표준입력 : int[] numbers
연산
- 인덱스 0, 합계 0 에서 시작한다.
- 현재 인덱스를 선택하는 경우, 선택하지 않는 경우 두 가지를 재귀함수로 실행한다. 그림으로 그려보자면 트리 모양으로 모든 경우의 수를 탐색하게 된다.
출력
- result 변수를 출력한다.
💡 코드
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
static int N;
static int S;
static int[] numbers;
static int result;
public static void main(String[] args) throws Exception {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer input = new StringTokenizer(br.readLine());
N = Integer.parseInt(input.nextToken());
S = Integer.parseInt(input.nextToken());
input = new StringTokenizer(br.readLine());
numbers = new int[N];
for (int i = 0; i < N; i++) {
numbers[i] = Integer.parseInt(input.nextToken());
}
br.close();
backTracking(0, 0);
if (S == 0) result--;
System.out.println(result);
}
static void backTracking(int idx, int sum) {
if (idx == N) {
if (sum == S) result++;
return;
}
// 현재 요소를 포함하는 경우
backTracking(idx + 1, sum + numbers[idx]);
// 현재 요소를 포함하지 않는 경우
backTracking(idx + 1, sum);
}
}
💡 시간 복잡도, 공간 복잡도
시간 복잡도 | 공간 복잡도 |
---|---|
O(n^2) | O(n) |