POJ 1821 单调队列优化DP
题意:
有一道线性篱笆由N个连续的木板组成。有K个工人,你要叫他们给木板涂色。每个工人有3个参数:L 表示 这个工人可以涂的最大木板数目,S表示这个工人站在哪一块木板,P表示这个工人每涂一个木板可以得到的钱。要注意的是,工人i可以选择不涂任何木板,否则,他的涂色区域必须是连续的一段,并且S[i]必须包含在内。 最后还有,每块木板只能被涂一次。
思路:
第一眼,水题~dp[i][j]表示第i个人刷的最后一面墙是j时的最大获利
一看数据范围,我水了。。。
怎么优化呢?
dp[i][j]含义同上
dp[i][j]=max(dp[i-1][k]+(j-k)*p[i]) j-l[i]+1<=k+1<=s[i]
dp[i][j]=max(dp[i-1][j],dp[i][j-1],dp[i][j])//第i个人不刷,第j面墙不刷
其中第二个方程转移时0(1)的,不用优化了,第一个方程转移时0(n)的,我们要想办法优化它
优化方程一般就是恒等变形+找规律
dp[i][j]=max(dp[i-1][k]-k*p[i])+j*p[i]
其中j*p[i]在i,j两重循环中相当于常数,所以,对于状态dp[i][j]只要维护dp[i-1][k]-k*p[i]的最大值即可
就用单调队列维护
单调队列维护过程:(转自lyd神犇的博客)
单调队列具体的做法是:最外层循环为i,首先把j=1~s[i]-1转移完(因为它不涉及第三个转移),然后把(j-l[i]<=k<=s[i]-1)的决策点的F[i-1,k]-p[i]*k依次入队建立“入队早晚时间递增,F[i-1,k]-p[i]*k的值递减”的单调队列,接下来循环j=s[i]~s[i]+l[i]-1,进行这三个转移(第三个转移只需要用队首元素),其中每次需要把队首超出长度限制的决策点出队;最后把剩下的到n循环完,只需要前两个转移。
自己的单调队列写错了,不停地WA,借鉴了LYD神犇的步骤,嘿嘿,ac了~
1 #include <cstdio> 2 #include <cstring> 3 #include <cstdlib> 4 #include <algorithm> 5 #include <iostream> 6 7 #define N 120 8 #define M 17000 9 10 using namespace std; 11 12 struct RE 13 { 14 int l,p,s; 15 }re[N]; 16 17 int q[M],l[N],r[N],dp[N][17000],n,k; 18 19 inline bool cmp(const RE &a,const RE &b) 20 { 21 return a.s<b.s; 22 } 23 24 void read() 25 { 26 for(int i=1;i<=k;i++) 27 scanf("%d%d%d",&re[i].l,&re[i].p,&re[i].s); 28 sort(re+1,re+1+k,cmp); 29 30 for(int i=1;i<=k;i++) 31 { 32 l[i]=max(0,re[i].s-re[i].l); 33 r[i]=min(n,re[i].s+re[i].l-1); 34 } 35 } 36 37 void go() 38 { 39 for(int i=0;i<=n;i++) dp[0][i]=0; 40 for(int i=1;i<=k;i++) 41 { 42 for(int j=0;j<re[i].s;j++) dp[i][j]=dp[i-1][j];//第i个粉刷匠不刷任何墙 43 int h=0,t=0; 44 for(int j=l[i],tmp;j<re[i].s;j++)//将dp[i-1]层的最优状态存入单调队列 45 { 46 tmp=dp[i-1][j]-j*re[i].p; 47 while(t>h&&dp[i-1][q[t-1]]-q[t-1]*re[i].p<=tmp) t--; 48 q[t++]=j; 49 } 50 for(int j=re[i].s,tmp;j<=r[i];j++) 51 { 52 while(t>h&&j-q[h]>re[i].l) h++;//弹出不在范围中的元素 53 dp[i][j]=max(dp[i-1][j],dp[i][j-1]); 54 dp[i][j]=max(dp[i][j],dp[i-1][q[h]]+(j-q[h])*re[i].p); 55 } 56 for(int j=r[i]+1;j<=n;j++) dp[i][j]=max(dp[i-1][j],dp[i][j-1]); 57 } 58 int ans=0; 59 for(int i=1;i<=n;i++) ans=max(ans,dp[k][i]); 60 printf("%d\n",ans); 61 } 62 63 int main() 64 { 65 while(scanf("%d%d",&n,&k)!=EOF) 66 { 67 read(); 68 go(); 69 } 70 return 0; 71 }