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

알고리즘/백준] 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); } }

728x90

from http://developer-hm.tistory.com/186 by ccl(A) rewrite - 2021-10-08 13:27:25