首先,对于答案输出为一个数字的题目,一般考虑用二分法,此题一样的道理,第一次的二分的范围为sum/n~sum,然后依次找mid,并判断,是否可以以mid为每月最大花费来将数据分组,并且使组数小于等于给定的组数,当组数小于m时,减少每月花费,当组数大于m时,加大每月花费。
此题注意一些细节:
1、组数可小于等于
2、当有单个数据直接大于了mid的时候,单独处理,即需要扩大每月花费
3、特别感谢POJ讨论版里的测试数据
# include<stdio.h>
//考虑少于月份完成,和边界问题
int a[100001];
int search(int n,int m,int low,int sum) {
int l,u,i,total,count,mid,flag,pos;
l=low;
u=sum;
while(l<=u) {//还是写成小于等于情况,避免出现错误
mid=(l+u)/2;
flag=1;
total=0;
count=0;
pos=0;
for(i=0; i<n; i++) {
if(a[i]==mid) {//有数据等于mid了,做一个标记
pos=1;
}
if(a[i]>mid) {//有数据大于mid了,直接跳出循环,把花费变大
l=mid+1;
flag=0;
break;
}
total+=a[i];
if(total>mid) {
i--;
count++;
total=0;
}
}
if(flag==0) {//直接进行下一次循环
continue;
}
if(total>0&&flag==1) {
count++;
}
if(pos==1&&count<=m) {//当有数据等于mid,并且此时分组数目小于给定组数,若不进行处理,则下一次循环的mid将更加小,显然不符合实际,所以选择直接跳出
return mid;
}
if(count<=m) {
u=mid-1;
} else {
l=mid+1;
}
}
return l;//选择返回下界
}
int main() {
int N,M,i,sum,low,ans;
while(scanf("%d %d",&N,&M)!=EOF) {
sum=0;
for(i=0; i<N; i++) {
scanf("%d",&a[i]);
sum+=a[i];
}
low=sum/N;
ans=search(N,M,low,sum);
printf("%d\n",ans);
}
}