修路-二分查找

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

24 3100 200 300 4009 4250 100 150 400 550 200 50 700 300

Sample Output

400900

代码如下：

#include <cstdio>  
#include <cstring>  
#include <cmath>  
#include <algorithm>  
  
using namespace std;  
int n,m,temp;  
int a[345];  
  
bool judge(int tim)  
{  
    int ret=0,team=1;  
    for(int i=0;i<n;i++){  
        if(ret+a[i]<=tim){  
            ret+=a[i];  
        }  
        else {  
            ret=a[i];  
            if(ret>tim)return 0;  
            team++;  
        }  
    }  
    if(team<=m) return 1;  
    if(team>m) return 0;  
}  
  
int bfind()  
{  
    int l=0,r=temp+1,mid;  
    while(l<r){  
        mid=l+(r-l)/2;  
        if(judge(mid))r=mid;  
        else l=mid+1;  
    }  
    return l;  
}  
int main()  
{  
    int T;  
    scanf("%d",&T);  
    while(T--){  
        memset(a,0,sizeof(a));  
        scanf("%d%d",&n,&m);  
        temp=0;  
        for(int i=0;i<n;i++){  
            scanf("%d",&a[i]);  
            temp+=a[i];  
        }  
        int ans=bfind();  
        printf("%d\n",ans);  
  
    }  
    return 0;  
}