[백준] 16401번 - 과자 나눠주기 [Java]

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

 

[백준] 16401번 - 과자 나눠주기 [Java]

[백준] 16401번 - 과자 나눠주기 [Java]

---

1.  아이디어

 

과자 길이에 대한 매개 변수 이분 탐색을 활용하면 해결할 수 있다.

---

2. 문제풀이

 

이분 탐색은 나눌 길이를 찾고 count 메서드는 찾은 길이로 줄 수 있는 과자 개수를 구하도록 구성했다.

같은 개수를 줄 수 있을 때 최대 길이를 구해야하므로 N개보다 적은 개수가 나오는 직전 길이를 구하는 Upper Bound로 계산 후 -1을 해서 N개가 나오는 최대 길이를 구하도록 했다. 이때 모든 조카에게 같은 길이의 막대과자를 나눠줄 수 없으면 0을 출력해야 하는데 이분 탐색에서 mid 값이 0이 되는 순간이므로 이때는 1을 반환하도록 조건 처리를 했다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int M = Integer.parseInt(st.nextToken());
        int N = Integer.parseInt(st.nextToken());

        int[] arr = new int[N];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        int maxLen = binarySearchUpperBound(arr, M);

        System.out.println(maxLen);
    }

    private static int binarySearchUpperBound(int[] arr, int target) {
        int low = 0;
        int high = Integer.MAX_VALUE;

        while (low < high) {
            int mid = low + (high - low) / 2;
            if (mid == 0) return mid;

            int cnt = count(arr, mid);

            if (cnt >= target) low = mid + 1;
            else high = mid;
        }

        return high - 1;
    }

    private static int count(int[] arr, int len) {
        int cnt = 0;
        for (int n : arr) {
            cnt += n / len;
        }
        return cnt;
    }

}

---

4. 후기