POJ 3104 Drying 二分

一、题目大意

求出晾干所有衣服的最短时间，每一件衣服每1秒自动晾干1个单位，如果被烘烤，则每秒晾干k的单位（注意被烘烤时不会有自动晾干的一个单位）

二、解题思路

我们对时间进行二分，最小值为0，最大值为晾干最慢的衣服的时间+1，对每一个二分的值mid，计算是否能够在mid下晾干，思路为先去掉所有小于mid的值，对大于mid的值，计算它除以(k-1)的值向上取整，将记录求和为count，将这个count与mid进行比较，如果k==1的话，那就不用计算每个衣服，只需判断是否存在大于mid的衣服即可。

三、代码

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll arr[100007], inf = 0x3f3f3f3f3f3f3f3f, maxt = 0;
int n, k;
void input()
{
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
    {
        scanf("%lld", &arr[i]);
        maxt = max(maxt, arr[i]);
    }
    scanf("%d", &k);
    sort(arr, arr + n);
    arr[n] = inf;
}
bool judge(ll mid)
{
    int next = lower_bound(arr, arr + n + 1, mid + 1) - arr;
    if (next != n && k == 1)
    {
        return false;
    }
    ll count = 0;
    for (; next < n; next++)
    {
        ll val = arr[next] - mid;
        if (val % (k - 1) == 0)
        {
            count += (val / (k - 1));
        }
        else
        {
            count += (val / (k - 1));
            count++;
        }
    }
    return count <= mid;
}
void binarySearch()
{
    ll left = 0, right = maxt + 1;
    while (left + 1 < right)
    {
        ll mid = (left + right) / 2;
        if (judge(mid))
        {
            right = mid;
        }
        else
        {
            left = mid;
        }
    }
    printf("%lld\n", right);
}
int main()
{
    input();
    binarySearch();
    return 0;
}