차곡차곡

[BOJ/Java] 백준 3040번 - 백설 공주와 일곱 난쟁이 본문

CS/Algorithm

[BOJ/Java] 백준 3040번 - 백설 공주와 일곱 난쟁이

sohy 2023. 8. 4. 09:25

백준 #3040 백설 공주와 일곱 난쟁이

 

3040번: 백설 공주와 일곱 난쟁이

매일 매일 일곱 난쟁이는 광산으로 일을 하러 간다. 난쟁이가 일을 하는 동안 백설공주는 그들을 위해 저녁 식사를 준비한다. 백설공주는 의자 일곱개, 접시 일곱개, 나이프 일곱개를 준비한다.

www.acmicpc.net

 

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

public class Main {
	static int[] nums = new int[9];
	static boolean[] isSelected = new boolean[9];
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		for (int i = 0; i < 9; i++) {
			nums[i] = Integer.parseInt(br.readLine());
		}
		
		setSum(0, 0);
	}
	
	public static boolean setSum(int cnt, int sum) {
		if (cnt == 7) {
			if(sum == 100 ) {
				for (int i = 0; i < 9; i++) {
					if (isSelected[i]) {
						System.out.println(nums[i]);
					}
				}
				return true;
			}
			return false;
		}
		
		for (int i = 0; i < 9; i++) {
			if (isSelected[i]) continue;
			isSelected[i] = true;
			if(setSum(cnt+1, sum+nums[i])) {
				return true;
			}
			isSelected[i] = false;
		}
		
		return false;
	}

}

원소가 7개이면서 합이 100이 되는 부분 집합을 구한다.

Comments