【树形dp】hdu 1011 Starship Troopers

题目链接

我的理解：dfs+dp=树形DP

【Key】建树+dfs+更新dp[i][j]

dp[i][j] : j个士兵进攻房间i可获得的最大概率,final 我们求dp[1][m]

水平有限，不明白为什么要建立无向图，有些细节还是不懂

#include<bits/stdc++.h>
using namespace std;

#define clr(k,v) memset(k,v,sizeof(k))
const int maxn=1e2+10;
vector<int>child[maxn];
int n,m,bug[maxn],v[maxn];
int dp[110][110];//dp[i][j]:j个士兵进攻房间i可获得的最大概率,final 我们求dp[1][m]
bool vis[110];

void dfs(int u){
int size=child[u].size();
vis[u]=true;
int cost=(bug[u]+19)/20;//当前节点需要花费多少士兵
for( int i=cost; i<=m; ++i ){
dp[u][i]=v[u]; //初始化过程
}
for(int i=0;i<size;++i){
if(!vis[child[u][i]]){
dfs(child[u][i]);
for(int j=m; j>=cost; --j ) //j个人攻打当前房间u,但要保证能拿下j>=cost
for( int k=1; j+k<=m; ++k )
if(dp[child[u][i]][k]) dp[u][j+k]=max(dp[u][j+k],dp[u][j]+dp[child[u][i]][k]);
}
}
}

int main()
{
// freopen("input.txt","r",stdin);
while(cin>>n>>m){
if( n==-1 && m==-1 ) break;
for(int i = 1; i <= n; ++ i)
scanf("%d%d",bug+i,v+i);//bug[]房间的障碍数bugs，v[]房间出现brain的概率
int a,b;
for(int i=1;i<n;++i){
cin>>a>>b;
child[a].push_back(b);
child[b].push_back(a);
}
clr(vis,false);
clr(dp,0);
dfs(1);
for(int i=1;i<=n;++i) child[i].clear();
if(m==0) {
puts("0");
continue;
}
printf("%d\n",dp[1][m]);
}
return 0;
}