lightoj 1048 - Conquering Keokradong 二分答案

給定n+1個數字，將這些數字組合爲k+1個數字（因爲都是從0開始的）...問最大的數字最小的情況是啥，並且要求排在前面的數字儘可能大。

二分答案，判定是否成立。

判定的時候考慮應該怎麼組合爲k+1個數字都小於x...後來想想真tm傻逼，如果組合出的數字都小於等於x且數量比k+1少是肯定成立的...

然後輸出的時候..少一個數字當時真惆悵..單步了一下看出來了。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define N 1010
using namespace std;
int n,k;
int a[N];
int pd(int x)
{
    int cnt=0,tmp=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]>x) return 0;
        if(tmp+a[i]<=x)
        {
            tmp+=a[i];
        }
        else
        {
            tmp=a[i];
            cnt++;
        }
    }
    return cnt<=k;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++)
    {
        scanf("%d %d",&n,&k);
        n++;
        int l=0,r=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            r+=a[i];
            l=max(l,a[i]);
        }
        int ans;
        while(l<=r)
        {
            int mid=(l+r)/2;
            if(pd(mid))
            {
                r=mid-1;
                ans=mid;
            }
            else
                l=mid+1;
        }
        printf("Case %d: %d\n",cas,ans);
        int tmp=0,cnt=0;
        for(int i=1;i<=n;i++)
        {
            if(tmp+a[i]>ans||n-i<k-cnt)
            {
                printf("%d\n",tmp);
                tmp=a[i];
                cnt++;
            }
            else
                tmp+=a[i];
        }
        printf("%d\n",tmp);
    }
    return 0;
}