알고리즘

백준 6236번 용돈 관리 (JAVA)

박카스마시며코딩 2022. 5. 18. 14:04

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

 

6236번: 용돈 관리

현우는 용돈을 효율적으로 활용하기 위해 계획을 짜기로 하였다. 현우는 앞으로 N일 동안 자신이 사용할 금액을 계산하였고, 돈을 펑펑 쓰지 않기 위해 정확히 M번만 통장에서 돈을 빼서 쓰기로

www.acmicpc.net

 

저는 해당 문제를 이분 탐색으로 해결하였습니다.

저는 인출할 수 있는 금액으로 m번의 인출만으로 해당 금액을 다 커버할 수 있는지를 확인하고 커버할 수 없다면 

start = mid +1 , 커버할 수 있다면 end = mid - 1로 초기화하면서 최소 인출금액을 찾았습니다.

 

시간복잡도 (nlog(m)) : n은 입력의 길이 m은 최대 인출 금액 (10_000 * 100_000)

 

package BOJ.binarysearch;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
import java.util.function.Function;

public class BOJ_6236 {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine()," ");
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(st.nextToken());
        int m = stoi.apply(st.nextToken());
        int[] money = new int[n];
        for(int i = 0 ; i < n ; i++){
            money[i] = stoi.apply(br.readLine());
        }
        int result = binarySearch(money,n,m);
        System.out.println(result);
    }
    private static final int MIN_MONEY = 1;
    private static final int MAX_MONEY = 10_000 * 100_000;
    private static final int INF = 987654321;
    private static int binarySearch(int[] money, int n, int m) {
        int result = MAX_MONEY;
        int start = MIN_MONEY;
        int end = MAX_MONEY;
        while(start <= end){
            int mid = (start + end) / 2;
            if(checkCnt(money,n,m,mid)){
                result = Math.min(result,mid);
                end = mid - 1;
            }else{
                start = mid + 1;
            }
        }
        return result;
    }

    private static boolean checkCnt(int[] money, int n, int m,int withdrawal) {
        int remainder = 0;
        int result = 0;
        for(int i = 0 ; i < n ; i++){
            if(remainder >= money[i]){
                remainder -= money[i];
                continue;
            }
            if(withdrawal >= money[i]){
                result++;
                if(result > m){
                    return false;
                }
                remainder = withdrawal;
                remainder -= money[i];
                continue;
            }
            if(withdrawal < money[i]){
                return false;
            }
        }
        return true;
    }
}