알고리즘 문제

[백준] 2805번 나무 자르기

feelcoding 2021. 2. 16. 21:12
728x90

www.acmicpc.net/problem/2805

 

2805번: 나무 자르기

첫째 줄에 나무의 수 N과 상근이가 집으로 가져가려고 하는 나무의 길이 M이 주어진다. (1 ≤ N ≤ 1,000,000, 1 ≤ M ≤ 2,000,000,000) 둘째 줄에는 나무의 높이가 주어진다. 나무의 높이의 합은 항상 M보

www.acmicpc.net

parametric search를 이용해서 풀었다.

#include <iostream>
#include <algorithm>
using namespace std;

int main() {
	static long long arr[1000005];
	int n;
	long long m;
	int maxHeight = 0;
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		cin >> arr[i];
		if (arr[i] > maxHeight) maxHeight = arr[i];
	}
	int start = 0;
	int end = maxHeight;
	long long mid;
	long long total = 0;
	while (start + 1 < end) {
		total = 0;
		mid = (start + end) / 2;
		for (int i = 0; i < n; i++) {
			total += max((long long)0, arr[i] - mid);
		}
		if (total == m) {
			cout << mid;
			return 0;
		}
		else if (total < m) {
			end = mid;
		}
		else if (total > m) {
			start = mid;
		}
 	}
	cout << start;
	return 0;
}
728x90