首页 > 代码库 > HLG 1039 修路 (二分+贪心)

HLG 1039 修路 (二分+贪心)

链接:http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1039

Description

前段时间,某省发生干旱,B山区的居民缺乏生活用水,现在需要从A城市修一条通往B山区的路。假设有A城市通往B山区的路由m条连续的路段组成,现在将这m条路段承包给n个工程队(≤ ≤ 300)。为了修路的便利,每个工程队只能分配到连续的若干条路段(当然也可能只分配到一条路段或未分配到路段)。假设每个工程队修路的效率一样,即每修长度为1的路段所需的时间为1。现在给出路段的数量m,工程队的数量n,以及m条路段的长度(这m条路段的长度是按照从A城市往B山区的方向依次给出,每条路段的长度均小于1000),需要你计算出修完整条路所需的最短的时间(即耗时最长的工程队所用的时间)。

Input

第一行是测试样例的个数T ,接下来是T个测试样例,每个测试样例占2行,第一行是路段的数量m和工程队的数量n,第二行是m条路段的长度。

Output

对于每个测试样例,输出修完整条路所需的最短的时间。

Sample Input

2

4 3

100 200 300 400

9 4

250 100 150 400 550 200 50 700 300

Sample Output

400

900


代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <map>
#include <vector>
#include <string>
#define MAXN 1005
#define RST(N)memset(N, 0, sizeof(N))
using namespace std;


int a[305], cas, n, m;
int tot, Max, sum;
int low, high, mid;


void Init()
{
    scanf("%d %d", &m, &n);
    Max = 0; tot = n-1;
    low = 1, high = 0;
    for(int i=1; i<=m; i++) {
        scanf("%d", &a[i]);
        high += a[i];
        if(a[i] > Max) Max = a[i];
    }
    low = Max, sum = a[1];
}


int main()
{
    scanf("%d", &cas);
    while(cas--) {
        Init();
        while(low < high) {
            mid = (low + high)/2;
            sum = a[1];
            for(int i=2; i<=m; i++) {
                if(a[i]+sum > mid) tot--, sum = a[i];
                else sum += a[i];
            }
            if(tot < 0) low = mid + 1;
            else high = mid;
        }
        printf("%d\n", low);
    }
    return 0;
}