传送门
hzw说是树形dp
扯淡
直接贪心。
用f[x]表示以x为根的子树内,离x最远的点的距离。
显然有f[x]=max(f[son]+dist[x][son])
按照贪心思想,在x与儿子之间放延迟器最好。
所以答案=sigma(f[x]-f[son]-dis[x][son])
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 500005
#define ll long long
using namespace std;
ll f[N],ans;
int head[N],n,s,x,y,z,tot;
struct edge{
int to,next;
ll val;
}e[N*2];
inline ll read(){
int x=0,f=1;
char ch=getchar();
while (ch<'0'||ch>'9'){
if (ch=='-') f=-1;
ch=getchar();
}
while (ch>='0'&&ch<='9'){
x=x*10+ch-48;
ch=getchar();
}
return x*f;
}
inline void ins(int x,int y,ll u){
e[++tot].to=y;
e[tot].val=u;
e[tot].next=head[x];
head[x]=tot;
}
void dfs(ll x,ll fa){
for (int i=head[x];i;i=e[i].next)
if (e[i].to!=fa){
dfs(e[i].to,x);
f[x]=max(f[x],f[e[i].to]+e[i].val);
}
for (int i=head[x];i;i=e[i].next)
if (e[i].to!=fa) ans+=f[x]-f[e[i].to]-e[i].val;
}
int main(){
n=read();
s=read();
for (int i=1;i<n;i++){
x=read();
y=read();
z=read();
ins(x,y,z);
ins(y,x,z);
}
dfs(s,0);
printf("%lld",ans);
}