zzuli D

D

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 187 Solved: 42

SubmitStatusWeb Board
Description

晴天想把一个包含n个整数的序列a分成连续的若干段，且和最大的一段的值最小，但他有强迫症，分的段数不能超过m段，然后他就不会分了。。。他想问你这个分出来的和最大的一段的和最小值是多少？

Input

第一行输入一个整数t，代表有t组测试数据。
每组数据第一行为两个整数n,m分别代表序列的长度和最多可分的段数。
接下来一行包含n个整数表示序列。
0<=n<=50000 1<=m<=n，0<=ai<=10000。

Output

输出一个整数表示和最大的一段的最小值。

Sample Input

1
3 2
1 3 5
Sample Output

5
HINT

1 3 5 分成一段可以为1 3 5和为9，分成两段可以为1，3 5或者1 3，5，和最大的一段值分别为8，5，所以答案为5

一道二分题比赛的时候没想到/(ㄒoㄒ)/~~

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>

using namespace std;
int a[50010];
int n,m;
bool judge(int mid){

    int s=0; int time=0;
    for(int i=1;i<=n;i++){
        if(a[i]>mid)
            return true;
        if(s+a[i]<=mid)
            s+=a[i];
        else{
            s=a[i];
            time++;
        }
        if(time>m-1)
            return true;
    }
    return false;
}
int main(){
    int T; scanf("%d",&T);
    while(T--){
        scanf("%d %d",&n,&m);
        int maxx=0; int sum=0;
        for(int i=1;i<=n;i++){
            scanf("%d", &a[i]);
            sum+=a[i];
            maxx=max(a[i],maxx);
        }
        int r=sum; int l=maxx;
        int ans=l;
        while( l <= r ){
            int mid=(l+r)>>1;

            if( judge( mid ) ){

                l=mid+1;
            }
            else{
                ans=mid;
                r=mid-1;
            }

        }
        printf("%d\n",ans);
    }
    return 0;
}