본문 바로가기
Problem Solving/Baekjoon

[백준] 17610 양팔저울 - Brute Force / Java

by graycode 2023. 7. 19.

 문제 링크

 

17610번: 양팔저울

무게가 서로 다른 k개의 추와 빈 그릇이 있다. 모든 추의 무게는 정수이고, 그릇의 무게는 0으로 간주한다. 양팔저울을 한 번만 이용하여 원하는 무게의 물을 그릇에 담고자 한다. 주어진 모든 추

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

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

public class Main {

    static int k;
    static int[] weights;
    static boolean[] measurable;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        k = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());

        weights = new int[k];
        int sum = 1;
        for (int i = 0; i < k; i++)
            sum += weights[i] = Integer.parseInt(st.nextToken());

        measurable = new boolean[sum];
        dfs(0, 0);

        int cnt = 0;
        for (int i = 1; i < sum; i++)
            if (!measurable[i]) cnt++;

        bw.write(String.valueOf(cnt));
        bw.flush();
    }

    private static void dfs(int weight, int idx) {
        if (idx == k) {
            if (weight > 0) measurable[weight] = true;
            return;
        }

        dfs(weight, idx + 1);
        dfs(weight + weights[idx], idx + 1);
        dfs(weight - weights[idx], idx + 1);
    }

}

댓글