题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1011
思路:很明显的树形背包
定义dp[root][m]表示以root为根,派m个士兵的最优解,那么dp[root][m]=max(dp[root][m],dp[root][k]+dp[son][j]) k+j<=m son为root 的孩子
树形dp的思路一般就是着父亲和孩子节点之间的转移关系,然后从dfs到叶子节点,然后开始从叶子节点向上进行DP
最重要的就是建树和dfs求解的过程,如果掌握了怎样建树和怎样进行DP,那么树形DP就会发现很简单了。。。。
加油!!!ACMer
代码如下:
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 110
int n,m;
int dp[MAX][MAX];
int tol;
class node
{
public:
int to;
int next;
};
node edge[MAX*];
int head[MAX];
int bugs[MAX];
int brain[MAX];
int vis[MAX];
void Build_Tree(int u,int v)
{
edge[tol].to=v;
edge[tol].next=head[u];
head[u]=tol++;
}
void init()
{
memset(head,-,sizeof(head));
memset(dp,,sizeof(dp));
memset(vis,,sizeof(vis));
memset(bugs,,sizeof(bugs));
memset(brain,,sizeof(brain));
tol=;
}
void dfs(int root)
{
vis[root]=;
int cost=(bugs[root]+)/;//important
for(int i=cost;i<=m;i++)
dp[root][i]=brain[root];
for(int i=head[root];i!=-;i=edge[i].next)
{
if(vis[edge[i].to]) continue;
int son=edge[i].to;
dfs(son); for(int j=m;j>=cost;j--)
{
for(int k=;k+j<=m;k++)
if(dp[son][k])
dp[root][j+k]=max(dp[root][j+k],dp[son][k]+dp[root][j]);
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==-||m==-) break;
init();
for(int i=;i<=n;i++)
scanf("%d%d",&bugs[i],&brain[i]);
for(int i=;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
Build_Tree(u,v);
Build_Tree(v,u);
}
if(m==) {cout<<""<<endl;continue;}
dfs();
cout<<dp[][m]<<endl; }
return ;
}