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的,只允许穿过一个的。