[알고리즘/백준] 2805 나무 자르기 - 자바(Java), 이분 탐색(Binary Search)

문제

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

풀이 코드

• 이분 탐색을 이용해서 풀이해야 시간 초과가 발생하지 않는다.
• 답을 구하는 과정에서 잘린 나무의 합이 int의 범위를 초과할 수 있기 때문에 잘린 나무의 합은 long형 변수로 사용한다.

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

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 n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());

        int[] trees = new int[n];
        int max = Integer.MIN_VALUE;

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            trees[i] = Integer.parseInt(st.nextToken());
            max = Math.max(max, trees[i]);
        }

        int left = 0;
        int right = max;
        int answer = 0;

        while(left <= right) {
            int mid = (left+right)/2;
            long sum = 0;

            for (int tree : trees) {
                if (tree - mid > 0) {
                    sum += (tree - mid);
                }
            }

            if (sum >= m) {
                left = mid + 1;
                answer = mid;
            } else {
                right = mid - 1;
            }
        }

        System.out.println(answer);
    }
}