好久没碰到这样的树形dp了,反而不会了
dfs两发的dp固然重要,简单的也不能遗忘啊
这题给你一棵树,然后边有权值,然后有k个点标记了,要让你花费最小,删除边,是的k个都互相不连通
可以想到状态肯定是dp[i][2],dp[i][0]表示到i节点,往下面,没有与标记的相连的最小花费
dp[i][1]表示i节点,往下面连着一个标记节点的最小花费
如果i是标记节点,那么dp[i][1]=INF
转移的话就是讨论一下i是否标记,子节点是否标记
还要记录一下子树里是否有标记节点,如果没有那么就continue
计算dp[i][1]的时候,如果子节点v是标记节点,那么dp[i][1]先加上dp[v][0]+edge[i].c,先做成不和标记节点相连,然后把一个子树替换成有标记节点的即可
这就要开一个变量tmp记录,v是标记节点的话,tmp=min(tmp,−edge[i].c)
如果v是非标记结点,那么dp[i][1]+=min(dp[v][0],dp[v][1]+edge[i].c)
tmp=min(tmp,dp[v][1]−min)
这一点要注意,居然二了,没注意到,然后错了
代码:
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")
using namespace std;
#define MAX 100005
#define MAXN 1000005
#define maxnode 205
#define sigma_size 2
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lrt rt<<1
#define rrt rt<<1|1
#define middle int m=(r+l)>>1
#define LL long long
#define ull unsigned long long
#define mem(x,v) memset(x,v,sizeof(x))
#define lowbit(x) (x&-x)
#define pii pair<int,int>
#define bits(a) __builtin_popcount(a)
#define mk make_pair
#define limit 10000
//const int prime = 999983;
const int INF = 0x3f3f3f3f;
const LL INFF = 0x3f3f;
//const double pi = acos(-1.0);
const double inf = 1e18;
const double eps = 1e-9;
const LL mod = 1e6+3;
const ull mx = 133333331;
/*****************************************************/
inline void RI(int &x) {
char c;
while((c=getchar())<'0' || c>'9');
x=c-'0';
while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
}
/*****************************************************/
struct Edge{
int v,next;
LL c;
}edge[MAX*2];
int head[MAX];
int tot;
int vis[MAX];
int num[MAX];
LL dp[MAX][2];
void init(){
mem(head,-1);
mem(vis,0);
tot=0;
}
void add_edge(int a,int b,int c){
edge[tot]=(Edge){b,head[a],c};
head[a]=tot++;
}
void dfs(int u,int fa){
if(vis[u]){
dp[u][1]=1e18;
dp[u][0]=0;
num[u]=1;
}
else{
dp[u][1]=dp[u][0]=0;
num[u]=0;
}
LL tmp=0;
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].v;
if(v==fa) continue;
dfs(v,u);
num[u]+=num[v];
if(num[v]==0){
continue;
}
if(vis[u]){
if(vis[v]) dp[u][0]+=dp[v][0]+edge[i].c;
else dp[u][0]+=min(dp[v][1]+edge[i].c,dp[v][0]);
}
else{
if(vis[v]){
dp[u][0]+=dp[v][0]+edge[i].c;
dp[u][1]+=dp[v][0]+edge[i].c;
tmp=min(tmp,-edge[i].c);
}
else{
dp[u][0]+=min(dp[v][0],dp[v][1]+edge[i].c);
LL k=min(dp[v][0],dp[v][1]+edge[i].c);
dp[u][1]+=k;
tmp=min(tmp,dp[v][1]-k);
}
}
}
dp[u][1]=dp[u][1]+tmp;
//cout<<dp[u][0]<<" "<<dp[u][1]<<" "<<u<<endl;
}
int main(){
//freopen("in.txt","r",stdin);
int t;
cin>>t;
while(t--){
int n,k;
cin>>n>>k;
init();
for(int i=1;i<n;i++){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add_edge(a,b,c);
add_edge(b,a,c);
}
for(int i=0;i<k;i++){
int a;
scanf("%d",&a);
vis[a]=1;
}
dfs(0,-1);
cout<<min(dp[0][0],dp[0][1])<<endl;
}
return 0;
}