您的位置:首页 > 其它

hdu 1561 (树形dp)The more, The Better

2013-04-12 14:02 507 查看
dp的思想方法,就是每个状态有很多个策略,然后怎么选择才是最优的。此时,假设它的所有后续状态的所有最优值已经计算出来。

那么,树形dp同样如此。

当到达一个节点i时,剩下j的容量,总共有x个子节点。假设子节点所有最优值已计算出来。而且已经算出来已经选择了k个子节点时的最优值。算选择第k+1的子节点时:(k+1的子节点有z个儿子,y表示i节点给k+1的子节点y的容量)

dp[i][k+1][j]=max{dp[i][k][j],dp[i][k][j-y]+dp[son[k+1]][z][y]}  (1<=y<j)

这个方程表示,给第k+1的子节点y的容量的最大值(此时,k+1子节点所有值已计算出了)加上给前k个节点j-y(此时,i节点的前k个节点所有值也已计算出来)的值是否比原值要大,然后更新。。。

然后,背包理解过的,都知道有个空间优化。j的值从后往前算,可以省掉n。因为总跟上一个节点的状态有关,那么,从后往前算,j减去一个值,还是属于上一个状态的值,。。

dp[i][j]=max{dp[i][j],dp[i][j-y]+dp[son[k]][y]}

总之,就是一个背包问题,每个节点的价值随着花费不同而改变!

#include <iostream>
#include<vector>
using namespace std;

int dp[205][205];       //dp[i][j]表示第i个节点有j的背包容量的最大值
vector<int> v[205];		//用来存放每个节点的子节点
int n,m,val[205];

int max(int a,int b)
{
return a>b?a:b;
}

void dfs(int x)
{
for(int i=0;i<v[x].size();i++)
{
dfs(v[x][i]);				//如果此节点有儿子,那么继续深搜
for(int j=m;j>=2;j--)		//此节点所有儿子节点做一次背包选择。算出dp[x][j]的最大值。此时,j是背包。
for(int k=1;k<j;k++)    //儿子节点的背包为j-k,父亲剩余k。  dp[x][j]=max{dp[x][j],dp[x][k]+dp[v[x][i]][j-k];
if(dp[x][k]!=-1&&dp[v[x][i]][j-k]!=-1)    //这里就是为了避免无谓的计算。因为对于任意一个节点的背包如果多于它整棵树的所有节点,就是浪费。
dp[x][j]=max(dp[x][j],dp[x][k]+dp[v[x][i]][j-k]);
}
}

int main()
{
//	freopen("in.txt","r",stdin);
int a;
while(scanf("%d %d",&n,&m)==2&&(m||n))
{
for(int i=0;i<=n;i++)
v[i].clear();               //每次都要把向量清空。。。谨记!!
for(int i=0;i<=n;i++)
for(int j=0;j<=m+1;j++)
dp[i][j]=-1;            //为什么初始化为-1呢,因为这样可以避免无所谓的计算
for(int i=1;i<=n;i++)
{
scanf("%d %d",&a,&val[i]);
v[a].push_back(i);
dp[i][1]=val[i];			//当第i个节点有容量为1时,即只能装它自己,所以值为自己的值。
}
m++;                  //把0加进去当做了根节点。所以背包容量要加1。
val[0]=dp[0][1]=0;     //根节点是加进去的,所以值为0
dfs(0);
m=m>n?n:m;           //如果总的背包容量大于总的节点数时,此时dp[0][m]=-1,所以此时总的背包容量为n。
printf("%d\n",dp[0][m]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  hdu dp树形dp