HDU 1011 Starship Troopers

Starship Troopers

樹形DP+分組揹包，根節點一定需要被攻佔。dp[i][j] : 在i節點派j個機器人，能獲得的最大概率。先不考慮根節點dp[i][j] = max{dp[i][j] , dp[i][j-k] + dp[u][k]} ,然後考慮i點也需要派兵攻佔。。在判斷一下就可了。具體參看程序

/*
    * author    : csuchenan
    * prog      : hdu 1011
    * Algorithm : Tree + DP , 注意 dp[i][j] ,當j小於對應的bugs[i]的時候應賦值爲0
    *           ：然後對於K==0 時，直接返回0
    * 14	csu_chenan	31MS	308K	1677B	C++
*/
#include <cstdio>
#include <cstring>
#include <vector>

#define maxn 105
using std::vector ;
int bugs[maxn] ;
int vals[maxn] ;
int dp[maxn][maxn]  ;
vector<int> G[maxn] ;
int n , K ;

void init(){
    for(int i = 0 ; i <= n ; i ++){
        G[i].clear() ;
    }
    memset(dp , 0 , sizeof(dp)) ;
}

bool read(){
    scanf("%d%d" , &n , &K) ;
    if(n == -1 || K == -1)
        return 0 ;
    init() ;
    for(int i = 1 ; i <= n ; i ++){
        scanf("%d%d" , &bugs[i] , &vals[i]) ;
        bugs[i] = (bugs[i] + 19) / 20 ;
    }
    int p ;
    int q ;
    for(int i = 1 ; i < n ; i ++){
        scanf("%d%d" , &p , &q) ;
        G[p].push_back(q) ;
        G[q].push_back(p) ;
    }
    return 1 ;
}

void dfs(int v , int f){
    for(int i = 0 ; i != G[v].size() ; i ++){
        int u = G[v][i] ;
        if(u == f)
            continue ;
        dfs(u , v) ;
    }

    for(int i = 0 ; i != G[v].size() ; i ++){
        int u = G[v][i] ;
        if(u == f)
            continue ;
        for(int j = K ; j >= 1 ; j --){
            for(int k = 1 ; k <= j ; k ++){
                if(dp[v][j] < dp[v][j - k] + dp[u][k]){
                    dp[v][j] = dp[v][j - k] + dp[u][k] ;
                }
            }
        }
    }
    for(int i = K ; i >= 0 ; i --){
        if(i >= bugs[v])
            dp[v][i] = vals[v]  + dp[v][i - bugs[v] ];
        else
            dp[v][i] = 0 ;
    }
}

void solve(){
    if(K == 0){
        printf("0\n") ;
        return ;
    }
    dfs(1 , 0) ;
    printf("%d\n" , dp[1][K]) ;
}

int main(){
    while(read()){
        solve() ;
    }
    return 0 ;
}