백준 1182번 : 부분수열의 합

반응형

https://www.acmicpc.net/problem/1182

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    int[] numArray;
    int n;
    int s;
    int count = 0;
    int[] resultArray;
    StringBuilder sb = new StringBuilder();

    public void solution() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        s = Integer.parseInt(st.nextToken());
        numArray = new int[n];
        st = new StringTokenizer(br.readLine());
        for(int i=0; i<n; i++){
            numArray[i] = Integer.parseInt(st.nextToken());
        }

        for(int i=0; i<n; i++){
            resultArray = new int[i+1];
            dfs(0, 0, i+1);
        }
        System.out.println(count);
    }

    public void dfs(int index, int depth, int size){
        if(depth==size){
            int result = 0;
            for(int a : resultArray){
                result += a;
            }
            if(result==s){
                count++;
            }
            return;
        }
        for(int i=index; i<n; i++){
            resultArray[depth] = numArray[i];
            dfs(i+1, depth+1, size);
        }
    }

    public static void main(String args[]) throws IOException {
        new Main().solution();
    }
   }

dfs를 활용한다. 모든 부분수열을 확인하고, 모두 더한 값이 주어진 s와 같을 경우 count 해주고 출력해준다.

 

반응형

'개발 > 알고리즘' 카테고리의 다른 글

백준 11726번 : 2xn 타일링  (0) 2023.01.27
백준 1463번 : 1로 만들기  (0) 2023.01.27
백준 11723번 : 집합  (0) 2023.01.25
백준 문제풀이 입력과 출력  (0) 2023.01.22
백준 6603번 : 로또  (0) 2023.01.20