POJ 3104 Drying 二分

一、题目大意

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

二、解题思路

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

三、代码

#include 
#include 
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;
}

你可能感兴趣的:(算法)