JZOJ 2678. 树B

题目

Description

已知无向连通图G由N个点，N-1条边组成。每条边的边权给定。现要求通过删除一些边，将节点1与另M个指定节点分开，希望删除的边的权值和尽量小，求此最小代价。

 

Input

每个输入文件中仅包含一个测试数据。

第一行包含两个整数N,M。

第二行至第N行每行包含3个整数,A、B、C，表示节点A与节点B有一条边相连，边权为C。

第N+1行至第N+M行每行包含一个整数X，表示要求与节点1分开的节点。

Output

输出文件仅包含一个整数，表示最小代价。

 

Sample Input

3 2
1 2 3
1 3 5
2
3

Sample Output

8

 

Data Constraint

 

 

Hint

对于20%的数据，2<=N<=10

对于100%的数据，2<=N<=10^6

 

分析

 

• 树形DP
• 设f[x]表示以x为根节点删除我需要点的最小值
• f[x]+=min(f[y],map[x][y])

 

代码

 

#include<bits/stdc++.h>
using namespace std;
struct node{
    int u,v,w,nx;
}g[1000011*2];
int cnt;
int li[1000001*2];
void add(int u,int v,int w) 
{
    g[++cnt].u=u; g[cnt].v=v; g[cnt].w=w; g[cnt].nx=li[u]; li[u]=cnt;
    g[++cnt].u=v; g[cnt].v=u; g[cnt].w=w; g[cnt].nx=li[v]; li[v]=cnt;
}
int flag[1000001];
int dp[1000001];
int b[1000001];
void dfs(int x)
{
    int ff=0;
    flag[x]=1;
    for (int i=li[x];i;i=g[i].nx)
    {
        int y=g[i].v;
        if (!flag[y])
        {
            if (b[g[i].v])
              ff+=g[i].w;
            else
            {
                dfs(y);
                ff+=min(dp[y],g[i].w); 
            }
        }
    }
    dp[x]=ff;
}
int main  ()
{
    int n,m;
    cin>>n>>m;
    for (int i=1,x,y,z;i<=n-1;i++)
    {
        cin>>x>>y>>z;
        add(x,y,z);
    }
    for (int i=1,x;i<=m;i++)
    {
        cin>>x;
        b[x]=1;
    }
    memset(dp,0x3f,sizeof(dp));
    dfs(1);
    cout<<dp[1];
}