poj 1505 Copying Books 二分

二分的时候左边的初始值弄错了,一直wa.

5 3
1 1 1 1 10
1 / 1 1 1 / 10
View Code
#include <stdio.h>

int m,n,k,a,ar[505];

void solve(int mid)
{
int i,j,s=0;
for (i=0;i<n;i++)
{
if(ar[i]+s<=mid)s+=ar[i];
else
{
s=ar[i];
k++;
}
}
}

void print()
{
int i,j,sum,cnt=1,pre[1000]={0};
for (i=n-2,sum=ar[n-1];i>=0;i--)
{
if(sum+ar[i]>a)
{
sum=ar[i];
pre[i]=1;
cnt++;
}
else sum+=ar[i];
}
for (i=0,j=cnt;j<m&&i<n;i++)
{
if(pre[i]==0)
{
pre[i]=1;
j++;
}
}
for (i=0;i<n-1;i++)
{
printf("%d ",ar[i]);
if(pre[i])printf("/ ");
}
printf("%d\n",ar[i]);
}

int main()
{
int T,i,j,l,r,mid;
scanf("%d",&T);
while (T--)
{
scanf("%d%d",&n,&m);
r=l=0;
for (i=0;i<n;i++)
{
scanf("%d",&ar[i]);
r+=ar[i];
if(l<ar[i])l=ar[i];
}
while (l<=r)
{
mid=(l+r)/2;
k=0;
solve(mid);
if(k<m)
{
r=mid-1;
a=mid;
}
else l=mid+1;
}
print();
}
return 0;
}


posted @ 2011-11-16 13:42  104_gogo  阅读(148)  评论(0编辑  收藏  举报