이번에 풀어본 문제는
백준 6236번 용돈 관리 입니다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
static int N,M;
static int [] input;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
M = Integer.parseInt(st.nextToken());
int start = 0;
input = new int[N];
for (int i = 0; i < N; i++) {
input[i] = Integer.parseInt(br.readLine());
start = Math.max(start, input[i]);
}
int end = 1_000_000_000; // 10,000 * 100,000
int mid;
int answer = Integer.MAX_VALUE;
while(start <= end) {
mid = (start + end) / 2;
int res = search(mid);
if (res <= M) {
answer = Math.min(answer, mid);
end = mid - 1;
}
else {
start = mid + 1;
}
}
System.out.print(answer);
}
static int search(int k) {
int count = 1;
int money = k;
for (int val : input) {
money -= val;
if (money < 0) {
count++;
money = k - val;
}
}
return count;
}
}
K만큼 M번 인출을하며 N일을 보낸다고 할 때, K값의 최소를 구하는 문제입니다.
이분탐색으로 K값을 mid로 하여 search 함수를 수행합니다.
search함수는 N일동안 사용할 금액으로 주어진 input 배열을 탐색하여 총 몇번의 인출이 필요한지 count값을 리턴해주고, 해당 값을 기준으로 start, end값을 조정하며 탐색을 계속 진행하면 됩니다.
마지막으로 최솟값이 담긴 answer을 출력해주면 해결할 수 있습니다.
이분탐색은 뭔가 오랜만에 푼거같은데, 처음 start, end값을 잘못 설정해서 꽤나 애먹었습니다..ㅠ