BZOJ 1044 二分+DP

第一问,明显的二分答案+验证。

第二问,dp[i][j]表示前j段切i刀的满足第一问的方案数,然后dp[i][j]=sigma(dp[i-1][k])  (k满足第一问限制)

显然在循环j的时候是可以算出来这个值的,所以总复杂度是n*m的。

其实我一开始写的是dp[i][j]表示前i个切j刀的方案数,显然这样写不能优化。。。

然后写出方程,发现调换i、j就可以了~

View Code
 1 #include <iostream>
 2 #include <cstring>
 3 #include <cstdlib>
 4 #include <cstdio>
 5 #include <algorithm>
 6 
 7 #define N 55555
 8 #define MOD 10007
 9 
10 using namespace std;
11 
12 int n,m,a[N],l,r,mid;
13 int sum[N],ans,ans1;
14 int dp[2][N],q[N];
15 
16 inline void read()
17 {
18     scanf("%d%d",&n,&m);
19     for(int i=1;i<=n;i++)
20     {
21         scanf("%d",&a[i]);
22         l=max(l,a[i]);
23         sum[i]=sum[i-1]+a[i];
24     }
25 }
26 
27 inline bool check()
28 {
29     int gs=m,p=1,sm;
30     while(gs!=0&&p<=n)
31     {
32         sm=0;
33         while(p<=n&&sm+a[p]<=mid) sm+=a[p++];
34         gs--;
35     }
36     if(p<=n&&sum[n]-sum[p-1]>mid) return false;
37     return true;
38 }
39 
40 inline void step1()
41 {
42     r=sum[n];
43     while(l<=r)
44     {
45         mid=(l+r)>>1;
46         if(check()) ans=mid,r=mid-1;
47         else l=mid+1;
48     }
49     printf("%d ",ans);
50 }
51 
52 inline void step2()
53 {
54     dp[0][0]=1;
55     for(int i=1;i<=m;i++)
56     {
57         int h=1,t=0,res;
58         q[++t]=0; res=dp[(i-1)&1][0];
59         for(int j=1;j<=n;j++)
60         {
61             while(h<=t&&sum[j]-sum[q[h]]>ans)
62             {
63                 res-=dp[(i-1)&1][q[h]]; h++;
64                 res%=MOD;
65                 if(res<0) res+=MOD;
66             }
67             dp[i&1][j]=res;
68             q[++t]=j; res+=dp[(i-1)&1][j];
69             res%=MOD;
70         }
71         for(int j=n-1;j>=1;j--)
72         {
73             if(sum[n]-sum[j]>ans) break;
74             ans1+=dp[i&1][j];
75             ans1%=MOD;
76         }
77         memset(dp[(i-1)&1],0,sizeof dp[(i-1)&1]);
78     }
79     printf("%d\n",ans1);
80 }
81 
82 inline void go()
83 {
84     step1();
85     step2();
86 }
87 
88 int main()
89 {
90     read(),go();
91     return 0;
92 }
没有人能阻止我前进的步伐,除了我自己!
原文地址:https://www.cnblogs.com/proverbs/p/2956796.html