백준 2805번 나무 자르기 (JAVA)

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

2805번: 나무 자르기

 

저는 이분탐색을 통해 문제를 해결하였습니다.

브루트포스로 할 경우 n*m의 시간복잡도이기 때문에 시간초과가 나올 것이라 판단하여 이분탐색을 통해 시간복잡도 n*log(m)으로 하여 문제를 풀었습니다.

 

 

package BOJ.binarysearch;

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

public class BOJ_2805 {

    public static void main(String[] args) throws Exception {
        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[] num = new int[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            num[i] = Integer.parseInt(st.nextToken());
        }
        long result = cal(num, n, m);
        System.out.println(result);
    }

    private static long cal(int[] num, int n, int m) {
        long start = 0;
        long end = 2_000_000_000;
        long result = 0;
        while (start <= end) {
            long mid = (start + end) / 2;
            long temp = check(num, mid, n);
            if (temp >= m) {
                result = mid;
                start = mid + 1;
            } else {
                end = mid - 1;
            }
        }
        return result;
    }

    private static long check(int[] num, long mid, int n) {
        long sum = 0;
        for(int i = 0 ; i < n ; i++){
            if(num[i] > mid){
                sum += num[i]- mid;
            }
        }
        return sum;
    }

}