bzoj 3743 [ Coci 2015 ] Kamp —— 树形DP

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3743

一开始想到了树形DP，处理一下子树中的最小值，向上的最小值，以及子树中的最长路和向上的最长路，就可以得到答案，可以DP；

然而写着写着写不下去了，不会求向上最小值和最长路；

于是看看TJ，原来要再记录一个次长路！

然而写挫了，写不下去了...

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const maxn=5e5+;
int n,m,hd[maxn],ct,fa[maxn],ans[maxn];
ll f[maxn][],g[maxn][],t[maxn],l[maxn][];
bool vis[maxn];
struct N{
    int to,nxt,w;
    N(int t=,int n=,int w=):to(t),nxt(n),w(w) {}
}ed[maxn<<];
void add(int x,int y,int z){ed[++ct]=N(y,hd[x],z); hd[x]=ct;}
void dfs(int x,int ft)
{
    fa[x]=ft;
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==ft)continue;
        dfs(u,x);
        vis[x]|=vis[u];
    }
}
void dp(int x)
{
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa[x])continue;
        dp(u);
        f[x][]+=(f[u][]+*vis[u]*ed[i].w);
        if(l[x][]<l[u][]+vis[u]*ed[i].w)
        {
            l[x][]=l[x][]; t[x]=u;
            l[x][]=l[u][]+vis[u]*ed[i].w;
        }
        else if(l[x][]<l[u][]+vis[u]*ed[i].w)l[x][]=l[u][]+vis[u]*ed[i].w;
    }
    f[x][]=f[x][]-l[x][];
}
void dp2(int x)
{
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa[x])continue;
        ans[x]=min(f[x][]+g[x][],f[x][]+g[x][]);
        g[u][]=g[x][]+f[x][]-f[u][];
        if(u==t[x])g[u][]=min(g[x][]+f[x][]-f[u][],g[u][]-l[x][])-ed[i].w;
        else g[u][]=min(g[x][]+f[x][]-f[u][],g[u][]-l[x][]);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=,x,y,z;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    for(int i=,x;i<=m;i++)
        scanf("%d",&x),vis[x]=;
    dfs(,); dp(); dp2();
    for(int i=;i<=n;i++)printf("%lld\n",ans[i]);
    return ;
}

囧

然后滚去看TJ，再次学习了一下优美的树形DP写法...

dis[x] 表示总路长，f[x][0] 表示子树中最长路，f[x][1] 表示子树中次长路，f[x][2] 表示向上最长路；

然后优美 dfs 即可...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const maxn=5e5+,inf=0x3f3f3f3f;
int n,m,hd[maxn],ct,siz[maxn];
ll dis[maxn],f[maxn][];
bool vis[maxn];
struct N{
    int to,nxt,w;
    N(int t=,int n=,int w=):to(t),nxt(n),w(w) {}
}ed[maxn<<];
void add(int x,int y,int z){ed[++ct]=N(y,hd[x],z); hd[x]=ct;}
void dfs(int x,int fa)
{
//    dis[x]=(vis[x]?0:-inf);
//    siz[x]=(vis[x]?1:0);
    siz[x]=vis[x]; dis[x]=;
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa)continue;
        dfs(u,x); siz[x]+=siz[u];
//        dis[x]+=dis[u]+ed[i].w;
        dis[x]+=dis[u]+(siz[u]?ed[i].w:);
    }
}
void dfs2(int x,int fa)
{
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa)continue;
        dis[u]=dis[x];//
        if(siz[u])dis[u]-=ed[i].w;
        if(siz[u]<m)dis[u]+=ed[i].w;//siz[u]==0
        dfs2(u,x);
    }
}
void dfs3(int x,int fa)
{
    f[x][]=f[x][]=(vis[x]?:(ll)-inf*inf);
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa)continue;
        dfs3(u,x);
        if(f[u][]+ed[i].w>f[x][])
        {
            f[x][]=f[x][];
            f[x][]=f[u][]+ed[i].w;
        }
        else if(f[u][]+ed[i].w>f[x][])f[x][]=f[u][]+ed[i].w;
    }
}
void dfs4(int x,int fa)
{
    for(int i=hd[x],u;i;i=ed[i].nxt)
    {
        if((u=ed[i].to)==fa)continue;
        if(f[u][]+ed[i].w==f[x][])f[u][]=f[x][]+ed[i].w;
        else f[u][]=f[x][]+ed[i].w;
        f[u][]=max(f[u][],f[x][]+ed[i].w);
        dfs4(u,x);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=,x,y,z;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    for(int i=,x;i<=m;i++)scanf("%d",&x),vis[x]=;
    dfs(,); dfs2(,); dfs3(,);
    f[][]=(vis[]?:(ll)-inf*inf); dfs4(,);
    for(int i=;i<=n;i++)printf("%lld\n",*dis[i]-max(f[i][],f[i][]));
    return ;
}