본문 바로가기
알고리즘/SWEA

[Java] SWEA 1486번 장훈이의 높은 선반

by 컴공맨 2021. 3. 4.
 

SW Expert Academy

SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!

swexpertacademy.com


풀이

부분집합을 이용하여 모든 조합을 만들고 B보다 크거나 같을 경우 그 차이가 가장 작은 값을 저장하여 해결했습니다.


코드

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

public class Solution {
	public static int N, B, answer;
	public static int[] people = new int[20];
	public static boolean[] isSelected = new boolean[20];
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder("");
		StringTokenizer st;
		
		int T = Integer.parseInt(br.readLine());
		for (int tc = 1; tc <= T; ++tc) {
			st = new StringTokenizer(br.readLine(), " ");
			N = Integer.parseInt(st.nextToken());
			B = Integer.parseInt(st.nextToken());
			
			st = new StringTokenizer(br.readLine(), " ");
			for (int i = 0; i < N; ++i) {
				people[i] = Integer.parseInt(st.nextToken());
			}
			
			answer = Integer.MAX_VALUE;
			powerset(0);
			
			sb.append("#").append(tc).append(" ").append(answer).append("\n");
		}
		
		System.out.println(sb);
	}
	
	public static void powerset(int cnt) {
		if (cnt == N) {
			int sum = 0;
			
			for (int i = 0; i < N; ++i) {
				if (isSelected[i]) {
					sum += people[i];
				}
			}
			
			if (sum >= B) {
				sum -= B;
				answer = (answer > sum) ? sum : answer;
			}
			
			return;
		}
		
		isSelected[cnt] = true;
		powerset(cnt + 1);
		isSelected[cnt] = false;
		powerset(cnt + 1);
	}
}

 

pyo7410/Algorithm

1일 1커밋을 목표로! Contribute to pyo7410/Algorithm development by creating an account on GitHub.

github.com

 

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

[Java] SWEA 1218번 괄호 짝짓기  (0) 2021.03.08
[Java] SWEA 1868번 파핑파핑 지뢰찾기  (0) 2021.03.05
[Java] SWEA 1249번 보급로  (0) 2021.03.03
[Java] SWEA 1211번 Ladder2  (0) 2021.03.02
[Java] SWEA 1226번 미로1  (0) 2021.03.01