【BZOJ1044】[HAOI2008]木棍分割 二分+DP

【BZOJ1044】[HAOI2008]木棍分割

Description

　　有n根木棍, 第i根木棍的长度为Li,n根木棍依次连结了一起, 总共有n-1个连接处. 现在允许你最多砍断m个连接处, 砍完后n根木棍被分成了很多段,要求满足总长度最大的一段长度最小, 并且输出有多少种砍的方法使得总长度最大的一段长度最小. 并将结果mod 10007。。。

Input

　　输入文件第一行有2个数n,m.接下来n行每行一个正整数Li,表示第i根木棍的长度.n<=50000,0<=m<=min(n-1,1000),1<=Li<=1000.

Output

　　输出有2个数, 第一个数是总长度最大的一段的长度最小值, 第二个数是有多少种砍的方法使得满足条件.

Sample Input

3 2

1

1

10

Sample Output

10 2

HINT

两种砍的方法: (1)(1)(10)和(1 1)(10)

题解：第一问直接二分答案即可，第二问显然是DP。

为了方便转移，我们需要对于每个位置i，求出pre[i]表示i的上一个砍断位置最远是多少，这个用双指针法和二分都可以搞。然后设f[i][j]表示在前i个连接处砍了j刀的方案数，那么用前缀和优化转移即可。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int P=10007;
int n,m,ans,len;
int v[50010],s[50010],f[2][50010],sf[2][50010],pre[50010];

inline int rd()
{
int ret=0,f=1;	char gc=getchar();
while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
bool solve(int x)
{
int i,sum=0,last=0;
for(i=1;i<=n&&sum<=m;i++)	if(s[i]-s[last]>x)
{
if(last==i-1)	return 0;
last=i-1,sum++;
}
sum++;
if(sum>m)	return 0;
return 1;
}
int main()
{
n=rd(),m=rd()+1;
int i,j,k,l=0,r=0,mid;
for(i=1;i<=n;i++)	v[i]=rd(),s[i]=s[i-1]+v[i],l=max(l,v[i]),r+=v[i];
while(l<r)
{
mid=(l+r)>>1;
if(solve(mid))	r=mid;
else	l=mid+1;
}
printf("%d ",len=r);
for(i=1;i<=n;i++)
{
l=0,r=i;
while(l<r)
{
mid=(l+r)>>1;
if(s[i]-s[mid]<=len)	r=mid;
else	l=mid+1;
}
pre[i]=r;
}
f[0][0]=1;
for(i=0;i<=n;i++)	sf[0][i]=1;
for(i=1;i<=m;i++)
{
k=(i&1);
sf[k][0]=0;
for(j=1;j<=n;j++)
{
f[k][j]=(sf[k^1][j-1]-((!pre[j])?0:sf[k^1][pre[j]-1])+P)%P;
sf[k][j]=(sf[k][j-1]+f[k][j])%P;
}
ans=(ans+f[k]
)%P;
}
printf("%d",ans);
return 0;
}