[BZOJ] 1044 [HAOI2008]木棍分割

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 4480  Solved: 1739
[Submit][Status][Discuss]
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,10
00),1<=Li<=1000.

Output
　　輸出有2個數, 第一個數是總長度最大的一段的長度最小值, 第二個數是有多少種砍的方法使得滿足條件.

Sample Input
3 2                           

1 

1

10
Sample Output
10 2
HINT
兩種砍的方法: (1)(1)(10)和(1 1)(10)

Source

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=50010;
const int M=10007;
int n,m,ans1,ans2,a[N];
int sum[N]={0},f[N]={0},pre[N],Sum[N]={0};
int main()
{
    int i,j,l=1,r=0;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;++i){
        scanf("%d",&a[i]);
        r+=a[i];l=max(l,a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    int num=0,use=0,now=1,mid;
    l-=1;
    while(1){
        num=use=0; 
        mid=(l+r)>>1;
        for(i=1;i<=n;++i){
            if(num+a[i]<=mid) num+=a[i];
            else num=a[i],use+=1;
        }
        if(use<=m) r=mid;
        else l=mid;
        if(r-l<=1) break;
    }
    ans1=r;
    for(i=1;i<=n;++i){
        if(sum[i]<=ans1) f[i]=1;
        else break;
    }
    for(i=1;i<=n;++i) 
      Sum[i]=(Sum[i-1]+f[i])%M;
    for(i=1;i<=n;++i){
        while(1){
            if(sum[i]-sum[now]<=ans1) break;
            else now+=1;
        }
        pre[i]=now-1;
    }
    for(i=1;i<=m;++i){
        for(j=1;j<=n;++j) 
          f[j]=(Sum[j-1]-Sum[pre[j]])%M;
        for(j=1;j<=n;++j)
          Sum[j]=Sum[j-1]+f[j];
        ans2=(ans2+f[n])%M;
    }
    printf("%d %d\n",ans1,ans2);
}