首页 > 代码库 > 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个工程队(n ≤ m ≤ 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;
}