BZOJ1044: [HAOI2008]木棍分割

Portal

首先第一问就直接二分了。没什么难度。
第二问比较容易想到的是：$f[i,j]$表示到第$i$个位置，切了$j$刀，并且在$i$后切了一刀的方案数。那么最后答案就是$\sum_{i=1}^{m+1}f[n,i]$
递推式:
$f[i,j]+=f[k,j-1]$（$j-1<=k<i$ 且$sum[i]-sum[k]<=ans$）ans为第一问答案。
显然$n^{2}m$是过不了的，空间也开不下。
空间优化还是比较容易的。发现$f[i,j]$只与$f[k,j-1]$有关，那么就可以用滚动数组优化一下，空间就不会被卡了。
然后考虑时间。
我们发现如果将$k$从$i-1$枚举到$j-1$，实际上在$sum[i]-sum[k]>ans$的时候就可以直接$break$掉了，因为$sum$数组有单调性，那么就可以二分了。然而$nmlgn$只能过$30$分。
然后又发现，其实不需要每次都重新计算$mink$,这是多余的，我们只需要提前预处理出每一个$i$对应的$mink[i]$就可以了，然后统计答案的时候，每次取$mink[i]$和$j-1$的较大值就可以了。显然如果$j-1>mink[i]$,必然满足$sum[i]-sum[j-1]<ans$。

【代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 50005
#define mod 10007
#define INF 0x7fffffff
using namespace std;
typedef long long ll;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,mx,ans,mn,Ans;
int a[N],f[N][2],sum[N],Sum[N],mnk[N];

void Input_Init()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) 
        a[i]=read(),mx+=a[i],mn=max(mn,a[i]),sum[i]=sum[i-1]+a[i];  
} 

bool Judge(int mid)
{
    int sum=0,num=0;
    for(int i=1;i<=n;i++)
    {
        if(sum+a[i]>mid) num++,sum=a[i];
        else sum+=a[i];
        if(num>m) return 0;
    }
    return 1;
}

void Solve()
{
    int l=mn,r=mx;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(Judge(mid)) ans=mid,r=mid-1;
        else l=mid+1;
    }
    printf("%d ",ans);
}

int Find(int x,int r)
{
    int rtn=0,l=0;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(sum[mid]>=x) rtn=mid,r=mid-1;
        else l=mid+1;
    } 
    return rtn;
}

void DP()
{
    f[0][0]=1;int now=0;for(int i=0;i<=n;i++) Sum[i]=1;
    for(int i=1;i<=n;i++) mnk[i]=Find(sum[i]-ans,i-1);
    for(int j=1;j<=m+1;j++)
    {
        now^=1;
        for(int i=0;i<=n;i++) f[i][now]=0;
        for(int i=j;i<=n;i++)
        {
            int t=max(j-1,mnk[i]);
            f[i][now]=(f[i][now]+Sum[i-1]-Sum[t-1])%mod;
        }
        Sum[0]=0;for(int i=1;i<=n;i++) Sum[i]=Sum[i-1]+f[i][now];
        Ans+=f[n][now];
    }
    printf("%d\n",Ans%mod);
    //printf("%d\n",(sizeof(f)+sizeof(sum)+sizeof(a))/1024/1024);
}

int main()
{
    Input_Init();
    Solve();
    DP();
    return 0;
}