题目大意:给定n个数字,需要把它最多分成k段,令ans是这k段中和最大的那个数,要使得k最小,求k并且输出k段合并后的数据;
题目解析:一开始一直在想这道题目应该是区间dp,区间dp确实可以用O(n^2)的方法把它求出来,但是之后输出k段的时候发现就不太对劲了,dp的过程不会算出来这个,所以很纠结;其实假设我们取定了ans,那么我们在合并的时候应该尽可能使得和接近ans,这样才是最优的,所以这里肯定会用到greedy的思想,想到这里不难想到这个ans应该二分出来了;
AC代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
int n,k,a[1010];
int jug(int pos)
{
int temp=0,cnt=0,i;
for(i=1;i<=n;i++)
{
if(a[i]>pos) return 0;
if(temp+a[i]>pos)
{
temp=a[i];
cnt++;
}
else
{
temp+=a[i];
}
}
return cnt<=k;
}
int main()
{
int cas,c,i,sum,ans;
scanf("%d",&cas);
for(c=1;c<=cas;c++)
{
scanf("%d%d",&n,&k);
sum=0;
n+=1;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
int l=0,r=sum;
while(l<=r)
{
int mid=(l+r)/2;
if(jug(mid)==0)
{
l=mid+1;
}
else
{
r=mid-1;
ans=mid;
}
}
printf("Case %d: %d\n",c,ans);
int temp=0,cnt=0;
for(i=1;i<=n;i++)
{
if(a[i]+temp>ans||k-cnt>n-i)
{
printf("%d\n",temp);
temp=a[i];
cnt++;
}
else
{
temp+=a[i];
}
}
printf("%d\n",temp);
}
return 0;
}