본문 바로가기
알고리즘

백준 s2 2805 나무 자르기 c++

by kyj0032 2024. 2. 17.

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

 

2805번: 나무 자르기

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

www.acmicpc.net

 

풀이

문제 설명과 풀이는 https://kyj0032.tistory.com/52와 유사하게 parametric search 썼음

 

조건(잘린 나무 합이 M 이상)에 맞는 절단기 높이 H의 최댓값을 구해야 함.

절단기 높이 H는 최대 10^9으로 1씩 내려가며 완전탐색을 할 수가 없음. 대신 이분탐색으로 높이 H를 탐색하며 조건에 맞는 최적해 찾기

 

정리하자면,

조건(잘린 나무 합이 M 이상)에 맞는 절단기 높이 H의 최댓값을 구하는 최적화 문제

-> 높이 H가 주어졌을 때 조건에 부합하는지 아닌지를 판단하는 결정 문제로 바꿔서 여러 번 탐색

 

절단기 높이-가져갈 수 있는 나무 함수는 감소함수이므로 parametric search 가능

 

코드

/*boj s2 2805 나무 자르기*/
#include <iostream>
#define MAXN 1000010
using namespace std;

int N, M;
long long tree[MAXN];

long long cut(long long x) {
    long long sum = 0;

    for (int i = 0; i < N; i++) {
        sum += (tree[i] - x) >= 0 ? (tree[i] - x) : 0;
    }

    return sum;
}

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin >> N >> M;
    for (int i = 0; i < N; i++)
        cin >> tree[i];

    long long st = 0;
    long long en = 1000000000;

    while (st < en) {
        long long mid = (st + en + 1) / 2;
        long long res = cut(mid);

        if (res < M) // 만족 X, 높이 더 내려가야 함
            en = mid - 1;
        else // 만족 O, 높이 최댓값을 구해야 하므로 값 올리기
            st = mid;
    }

    cout << st << "\n";
}

'알고리즘' 카테고리의 다른 글

백준 g5 18869 멀티버스 II c++  (0) 2024.02.20
백준 g5 2467 용액 c++  (0) 2024.02.19
백준 s2 16401 과자 나눠주기 c++  (0) 2024.02.17
백준 s4 1822 차집합 c++  (0) 2024.02.17
백준 s5 10815 숫자 카드 c++  (0) 2024.02.17