BZOJ3743 : [Coci2014]Kamp

d[x][0]表示x点向下走且回到x点的最少代价

d[x][1]表示x点向下走但不回到x点的最少代价

d[x][2]表示x点向下走的最长路

d[x][3]表示x点向下走的次长路

u[x][0]表示x点向上走且回到x点的最少代价

u[x][1]表示x点向上走但不回到x点的最少代价

一遍树形DP即可

ans[i]=min(d[i][0]+u[i][1],d[i][1]+u[i][0])

#include<cstdio>
#define N 500010
typedef long long ll;
int n,k,i,j,x,y,z,g[N],nxt[N<<1],v[N<<1],w[N<<1],ed,f[N],dis[N],h,t,q[N],size[N],de[N];
ll d[N][4],u[N][2],tmp;
bool is[N];
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline void add(int x,int y,int z){v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;}
inline ll min(ll x,ll y){return x<y?x:y;}
int main(){
  read(n),read(k);
  for(i=1;i<n;i++)read(x),read(y),read(z),add(x,y,z),add(y,x,z);
  while(k--)read(x),is[x]=1;
  q[h=t=1]=1;
  while(h<=t)for(j=g[x=q[h++]];j;j=nxt[j])if(v[j]!=f[x])f[q[++t]=v[j]]=x,dis[v[j]]=w[j];
  for(i=n;i;i--){
    size[x=q[i]]=is[x];
    for(j=g[x];j;j=nxt[j])if(v[j]!=f[x]&&size[v[j]]){
      size[x]+=size[v[j]];
      d[x][0]+=d[v[j]][0]+2LL*w[j];
      tmp=d[v[j]][1]-d[v[j]][0]-w[j];
      if(tmp<d[x][2])d[x][3]=d[x][2],d[x][2]=tmp,de[x]=v[j];else d[x][3]=min(d[x][3],tmp);
    }
    d[x][1]=d[x][0]+d[x][2];
  }
  for(i=2;i<=n;i++)if(size[1]>size[x=q[i]]){
    u[x][0]=u[f[x]][0]+d[f[x]][0]-d[x][0];
    u[x][1]=u[f[x]][1]+d[f[x]][0];
    if(de[f[x]]==x)u[x][1]=min(u[x][1],u[f[x]][0]+d[f[x]][0]+d[f[x]][3]);
    else u[x][1]=min(u[x][1],u[f[x]][0]+d[f[x]][1]);
    u[x][1]-=d[x][0]+dis[x];
    if(!size[x])u[x][0]+=2*dis[x],u[x][1]+=2*dis[x];
  }
  for(i=1;i<=n;i++)printf("%lld
",min(d[i][0]+u[i][1],d[i][1]+u[i][0]));
  return 0;
}