模板 树形dp

Posted 2022-08-13 xiaobuxie

 1 void dfs(int x)
 2 
 3     dp[x][1]=s[i],size[x]=1;
 4     for(int i=first[x];i;i=next[i])
 5     
 6         // first[x]相当于边,to[i]是x的子节点
 7         dfs(to[i]);
 8         for(int j=cmin(m,size[x]);j;j--)
 9         //这里size是to[i]之前的,
10             for(int k=1;k<=cmin(m-j,size[to[i]]);k++)
11             // k是指给多少个给当前子节点
12                 dp[x][j+k]=cmax(dp[x][j+k],dp[x][j]+dp[to[i]][k]);
13         size[x]+=size[to[i]];
14     
15 
16 //size[x]表示以i为根节点的子树中的点数，上面是主动更新
17 
18 void dfs(int x)
19 
20     size[x]=1;
21     for( int i=first[x];i;i=next[i])
22     
23         dfs(to[i]),size[x]+=size[to[i]];
24         for(int j=min(m,size[x]);j;j--)
25             for(int k=1;k<=min(j,size[to[i]]);k++)
26                 dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[to[i]][k]);
27     
28     for(int i=min(m,size[x]);i;i--) dp[x][i]=dp[x][i-1]+s[x];//保证父亲节点被选到
29