BOJ - 2805 ) 나무자르기

 

 

---

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

2805번: 나무 자르기

 

이분탐색으로 나무를 자르는 문제이다.

 

절단기의 H를 설정해주면 나무배열에서 H이상의 값을 가진나무들과 H의 차만큼 가져갈수있는 나무가 생긴다.

 

그때에, 필요한 나무를 가져가면서 설정할수있는 H의 최댓값을 찾는문제이다.

 

이분탐색으로 LEFT = 0 , RIGHT = *max_element()를 사용하여서 백터에서 가장 큰 값을 가지고 시작하였다.

 

처음에 LEFT를 나무의 가장 작은 *min_element()로 설정하니 자꾸 틀렸다고 나와서 생각해보니

 

필요한 나무길이가 많아서 최소작은 나무보다 더 아래쪽으로 절단기를 설정해주는 경우가 있었다.

 

 

---

#include <iostream>
#include <algorithm>
#include <vector>
 
using namespace std;
long long N , M , mid, cut_sum, ans;
vector <long long> wood;
void input() {
    cin >> N >> M ;
    for (int index = 0; index < N; index++) {
        long long t; cin >> t;
        wood.push_back(t);
    }
}
 
long long cut(vector< long long> wood, int mid) {
    long long sum = 0;
    for (auto wood_one : wood) {
        if (wood_one > mid) {
            sum += (wood_one - mid);
        }
    }
    return sum;
}
 
int main(void) {
    input();
 
    long long left = 0;
    long long right = *max_element(wood.begin(), wood.end());
 
    while (left <= right) {
 
        mid = (left + right) / 2;
        cut_sum = cut(wood, mid);
        
        if (cut_sum < M) {
            right = mid - 1;
        }
        else {
            if (ans < mid) ans = mid;
            left = mid + 1;
            
        }
 
        if (cut_sum == M) break;
    }
    cout << ans;
    return 0;
}