[백준 2805번 C/C++] 나무 자르기

 

 

목차

 

 

---

 

 

 

[백준 2805번 C/C++] 나무 자르기

 

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

2805번: 나무 자르기

 

---

 

해결전략

 

이분탐색

---

 

코드

 

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

int n, cnt;
long long m, maxHeight=0;

vector<long long> v;

long long Cal(int x)
{
   long long h = 0;
   
   if (x == 0) return h;

   for (int i = 0; i < n; i++) 
    {
        if(v[i]>=x)
            h += v[i] - x;
    }

    return h;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> m;
    v.resize(n);
    for (int i = 0; i < n; i++) {
        cin >> v[i];
    }

    long long lo = 0;
    long long hi = *max_element(v.begin(), v.end());
    long long mid = 0;

    while (lo <= hi)
    {
        mid = (lo + hi) / 2;

        if (Cal(mid) >= m) {
            lo = mid + 1;

            if (maxHeight < mid) maxHeight = mid;
        }
        else {
            hi = mid - 1;
        }
    }

    cout << maxHeight;

    return 0;
}

 

mid가 0이 나오는 경우가 생겨  Division Zero가 될 수 있다.

이런 상황을 방지하기 위해

maxHeight = 0;  //기본값 설정.

 if ( x == 0 ) return h;

long long mid = 0; //기본값 설정.

다음과 같은 안전장치를 만들어야 한다. 

---

 

 

유사문제

 

https://designerd.tistory.com/entry/%EB%B0%B1%EC%A4%80-1654%EB%B2%88-CC-%EB%82%98%EC%9D%B4%EC%88%9C-%EC%A0%95%EB%A0%AC

[백준 1654번 C/C++] 랜선 자르기