bzoj1044(木棍分割 也算不上单调队列优化dp..)

题目

显然第一问最大值最小化 二分最大值mx。第二问切割方式显然就是dp.
dp[i][j]=sigma(dp[k][j-1]) (j-1<=k<i sum[i]-sum[k]<=mx)
我就差一点点T了 T_T

//    Time:7792 ms    Memory:2264 kb
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=5e4+5,mod=1e4+7;
int a[N],sum[N],n,m;
inline int check(int mid){//最长一段<=mid 最少需要分几段
    int num=0,lastsum=0;
    for(int i=1;i<=n;++i){
//        if(sum[n]-sum[i-1]>mid*(m-num)) return 0; //是错的 因为i,i+1..可能和前面一组 所以就不该是sum[n]-sum[i-1]
        if(lastsum+a[i]>mid) ++num,lastsum=a[i];
        else lastsum+=a[i];
        if(sum[n]-sum[i-1]>(ll)mid*(m-num));//剪枝
    }
    return (++num)<=m;
}
int dp[N][2],q[N];//add(x,y),保证x y都为(-mod,mod);
inline void add(int &x,int y){//要是改成 x=((x+y)%mod+mod)%mod 就T了
    if(x+y<0) x=x+y+mod;
    else x=(x+y>mod)?(x+y-mod):(x+y);
}
int main(){
    scanf("%d%d",&n,&m),++m;
    int mx=0;
    for(int i=1;i<=n;++i)
        scanf("%d",&a[i]),mx=max(mx,a[i]),sum[i]=sum[i-1]+a[i];
    int low=mx,high=sum[n],mid;
    while(low<=high){
        mid=(low+high)>>1;
        if(check(mid)) high=mid-1;
        else low=mid+1;
    }
    printf("%d ",mx=high+1);
    int ans=0,flag=0;//flag 采用滚动数组优化
    for(int i=1;i<=n;++i) dp[i][flag^1]=(sum[i]<=mx);//这是前i个数字分成一段的dp值.
    add(ans,dp[n][flag^1]);
    for(int j=2;j<=m;++j){
        int head=1,tail=0,tmpans=0;flag^=1;//tmpans为了统计一段合法的dp[k][j-1]值的和.
        for(int i=1;i<=n;++i){
            while(head<=tail&&sum[i]-sum[q[head]]>mx) add(tmpans,-dp[q[head++]][flag]);//将这个决策点q[head]去掉 tmpans也要减去dp[q[head]][j-1]的值.
            dp[i][flag^1]=tmpans;
            if(i>=j-1) q[++tail]=i,add(tmpans,dp[i][flag]);
        }//将i这个决策点入队 对>=i+1以后的dp[i][j]转移有用.
        add(ans,dp[n][flag^1]);
    }
    printf("%d\n",ans);
}