hdu4313_题解

Brief Description：

给出一棵树，树边有权值，规定树上某些点为危险点，现在要去除一些边，使得危险点之间的路径不存在，要求删除的边的权值和最小，求这个权值和。

Analysis:

树型DP，在DFS里回溯到某个节点u时，保证

条件1.以这个节点为根的子树当中，危险点个数<=1;

dp[u]表示 保证条件1时 u连到子树中危险点（危险点最多只有一个）的路径的最短的一条边的权值，若u为危险点，dp[u]=无穷大

sum[u]表示 要使以u为根的子树满足条件1所需要删除的最小边值。

整棵树以标号为0的节点为根节点，最后的答案是sum[0]

具体见以下代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>

using namespace std;
typedef long long LL;
const int maxn = 101000;
const LL INF = 999999999999LL; //important!!
struct node{
    int id;
    LL w;
    node(){}
    node(int _id,LL _w){
        id = _id;   w = _w;
    }
};
LL dp[maxn],sum[maxn];
vector<node>V[maxn];
int n,m;
bool bb[maxn],vis[maxn];

void solve(int x){
    if(vis[x]) return;
    vis[x] = 1;

    int i,Size;
    Size = (int)V[x].size();
    if(Size == 1 && x != 0){
      if(bb[x]) dp[x] = INF;
      sum[x] = 0;
      return;
    }
    LL maxv = 0;
    for(i=0; i<Size; i++){
       int id = V[x][i].id;
       if(!vis[id]){
           solve(id);
           maxv = max(maxv,min(dp[id],V[x][i].w));    sum[x] += (sum[id]+min(dp[id],V[x][i].w));
       }
    }
    if(bb[x]) dp[x] = INF;
    else { dp[x] = maxv;  sum[x] -= maxv; }
}
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--){
        int i;
        scanf("%d%d",&n,&m);
        for(i=0; i<n; i++) V[i].clear();
        for(i=1; i<n; i++){
           int x,y;   LL w;
           scanf("%d%d%I64d",&x,&y,&w);
           V[x].push_back(node(y,w));   V[y].push_back(node(x,w));
        }
        memset(bb,0,sizeof(bb));
        while(m--){
           int x;
           scanf("%d",&x);  bb[x] = 1;
        }
        memset(dp,0,sizeof(dp));
        memset(vis,0,sizeof(vis));
        memset(sum,0,sizeof(sum));
        solve(0);
        printf("%I64d\n",sum[0]);
    }
    return 0;
}

// 2012.9.2观察体会：熟悉这种建模，dp=inf的，只允许穿过一个的。