【树形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;
}