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;
}