hrbust 1039 修路(二分+贪心 or DP)

题意是给一段从A到B的路,分成n段,由m个工程队来承包。且每个队伍只能承包连续的一段。

求最短的达成时间。


可以用DP做,但用二分更快。


二分的上界:当只有一个工程队来承包的时候,做工的完成时间为全部长度相加。

二分的下界:当有n个工程队来承包的时候,做工的完成时间为全部长度中的最大值。

找到上下界后,在二分中判断mid是否合题意,就ok了。


代码:

#include <stdio.h>

const int Maxn = 301;

int road[Maxn];
int m, n;

int check(int u)
{
    int cnt = 1;//
    int sum = 0;
    for (int i = 0; i < m; i++)
    {
        if(sum + road[i] <= u)
        {
            sum += road[i];
        }
        else
        {
            sum = road[i];
            cnt++;
            if(cnt > n)
                return -1;
        }
    }
    return -2;
}

int find(int low, int high)
{
    while (low < high)
    {
        int mid = (low + high) >> 1;
        //printf("---low--%d---high--%d---mid--%d\n", low, high, mid);
        int flag = check(mid);
        if(flag == -1)
            low = mid + 1;
        else if(flag == -2)
            high = mid;
    }
    return low;
}

int main()
{
    //freopen("in.txt", "r", stdin);
    int ncase;
    scanf("%d", &ncase);
    while (ncase--)
    {
        int downroad = -1;
        int uproad = 0;
        scanf("%d%d", &m, &n);
        for (int i = 0; i < m; i++)
        {
            scanf("%d", &road[i]);
            uproad += road[i];
            downroad = downroad > road[i] ? downroad : road[i];
        }
        printf("%d\n", find(downroad, uproad));
    }
    return 0;
}


你可能感兴趣的:(hrbust 1039 修路(二分+贪心 or DP))