今天学习了树形DP,用这道练练手
传送门 初拿到题有些棘手,因为不知道如何表示状态,再仔细一推,发现与树形背包相似,所以我们设dp[now][k] 表示以now为根结点时用了time分钟能获得的最大画数 如果now是展室,设s为now的画数,k为选取的画数,那么dp[now][5k]= ∑ k = 1 s d p [ n o w ] [ 5 ∗ k − 5 ] \sum_{k=1}^s{dp[now][5*k-5 ]} ∑k=1sdp[now][5∗k−5] 其中,time+5k<=m 即:
for(int i=1;i<=a[now].val;i++) { int s=tt+i*5; if(s>m) { break; } dp[now][s]=dp[now][s-5]+1; }而如果now是有两条分叉的走廊,我们可以把它抽象成一个没有画的房间,先遍历它的左子树和右子树,递归得到答案后,设左子树花费i秒偷盗,右子树花费j秒偷盗,则:
for(int i=0;i<=m-tt;i++){ for(int j=0;j<=m-tt-i;j++) { dp[now][tt+i+j]=max(dp[now][tt+i+j],dp[l][tt+i]+dp[r][tt+j]); } }输入时,还是用递归,如果是分叉走廊,则设这三条走廊分别为x(当前编号),2x,2x+1保证遍历时的有序 完整代码:
#include<cstdio> #include<iostream> using namespace std; int n,dp[605][605],m; struct ren{ int t,val; }a[1005]; void init(int x) { scanf("%d%d",&a[x].t,&a[x].val); a[x].t*=2;//最后要返回入口,每个走过的边要往返走一次 if(!a[x].val)//是分叉走廊 { init(x*2); init(x*2+1); } } void dfs(int now,int tt) { if(a[now].val) { for(int i=1;i<=a[now].val;i++) { int s=tt+i*5; if(s>m) { break; } dp[now][s]=dp[now][s-5]+1; } } else{ int l=now*2,r=now*2+1; dfs(l,tt+a[l].t);//遍历左子树 dfs(r,tt+a[r].t);//遍历右子树 for(int i=0;i<=m-tt;i++){ for(int j=0;j<=m-tt-i;j++) { dp[now][tt+i+j]=max(dp[now][tt+i+j],dp[l][tt+i]+dp[r][tt+j]);//时间为左子树所用时间+右子树所用时间 } } } } int main() { scanf("%d",&m); m-=1;//当你刚好走到入口时警察来了也跑不了 init(1); dfs(1,a[1].t);//从第一个走廊进入,至少都要花第一个走廊的时间 printf("%d\n",dp[1][m]); return 0; }附赠:偷天换日