[백준/BOJ] 15953번: 상금 헌터 (JAVA)

문제 링크:

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

15953번: 상금 헌터

 

소스 코드:

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

public class Main {

	/* 제1회 코드 페스티벌 정보 */
	int[] rank1Info  = {1, 2, 3, 4, 5, 6};
	int[] prize1Info = {500, 300, 200, 50, 30, 10};
	
	/* 제2회 코드 페스티벌 정보 */
	int[] rank2Info  = {1, 2, 4, 8, 16};
	int[] prize2Info = {512, 256, 128, 64, 32};
	
	public static void main(String[] args) throws IOException {
		Main solution = new Main();
		
		solution.solve();
	}
	
	public void solve() {
		try {
			BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
			BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
			
			StringTokenizer st = new StringTokenizer(br.readLine());
			
			// 등수 가정을 돌릴 횟수
			int t = Integer.parseInt(st.nextToken());
			
			for (int i = 0; i < t; i++) {
				st = new StringTokenizer(br.readLine());
				
				int rank1 = Integer.valueOf(st.nextToken());
				int rank2 = Integer.valueOf(st.nextToken());
				
				int result = calc(rank1, rank1Info, prize1Info) 
						   + calc(rank2, rank2Info, prize2Info);
				
				bw.append(result + System.lineSeparator()); // '/r/n'과 같은 역할
			}
			
			bw.flush();

			bw.close();
			br.close();
			
		} catch (NumberFormatException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	public int calc(int p, int[] rankInfo, int[] prizeInfo) {
		// 총 상금
		int totPrize  = 0;
		// 상금을 받을 수 있는 인원
		int totPerson = Arrays.stream(rankInfo).sum();

		if (p > 0 && p <= totPerson) {
			int count = 0;
			
			for (int i = 0; i < rankInfo.length; i++) {
				count += rankInfo[i];
				
				if (p <= count) {
					totPrize += prizeInfo[i];
					
					return totPrize * 10000;
				}
			}
		}
		return totPrize * 10000;
	}
}

각각의 기능을 하는 클래스를 따로 만들어서 구현하려고 했다.
이후에 페스티벌 정보가 더 추가되어도 쉽게 계산할 수 있게끔 정보를 담는 객체가 있으면 좋을 것 같다.