JZOJ5906【NOIP2018模拟10.15】传送门

       真.传送门

分析

　　　很明显这是道树形dp。

　　如果没有传送门的话，树上的距离之和就是边权之和的2倍，但是可以用传送省掉某些边，所以开数组dp[i][1/0]表示在i号结点放/不放传送门能省掉的最大权值。

　　通过分析可得，对于以任意一个结点为根的子树，如果把传送门放在这个结点上的话，那么与ta相连的每个点都可以直接传送会来，所以dp[u][1]+=max(dp[v][0]+edge[i].val,dp[v][1])。

　　如果不放的话，就只能把与这个结点的相连的最长子链给省掉，并且那条子链也不能放传送门（不然回不去了），

所以dp[u][0]=max(dp[u][0],dp[v][0]+edge[i].val)。

 

code 

#include<bits/stdc++.h>
#define maxn 1000010
#define ll long long
using namespace std;
int n,head[maxn],k;
ll sum,dp[maxn][2];
struct node
{
    int to,nxt;
    ll val;
} edge[maxn*2];
void add(int u,int v,ll w)
{
    edge[++k].nxt=head[u];
    edge[k].to=v;
    edge[k].val=w;
    head[u]=k;
}
void dfs(int u,int fa)
{
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v==fa) continue ;
        dfs(v,u);
        dp[u][1]+=max(dp[v][0]+edge[i].val,dp[v][1]);
        dp[u][0]=max(dp[u][0],dp[v][0]+edge[i].val);
    }
}
int main()
{
    scanf("%d",&n);
    int x,y;
    ll z;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%lld",&x,&y,&z);
        add(x,y,z),add(y,x,z);
        sum+=z*2;
    }
    dfs(1,1);
    printf("%lld\n",sum-max(dp[1][0],dp[1][1]));
    return 0;
}

转载于:https://www.cnblogs.com/wCTSd/p/9805152.html