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
,nxt[N<<1],v[N<<1],w[N<<1],ed,f
,dis
,h,t,q
,size
,de
;
ll d
[4],u
[2],tmp;
bool is
;
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\n",min(d[i][0]+u[i][1],d[i][1]+u[i][0]));
return 0;
}