这道题其实就是一个简单的树形DP。
我们可以发现,如果被使用道具的边离\(root\)越近,其影响范围就越大。所以,我们可以以\(DP[i]\)来表示以\(i\)节点为根的子树全部时态同步的最小花费。从底下来看,如果多点为根的子树都已经完成时态同步,这时要将这几棵子树时态同步,又自然要让他们的边权一样,直接统计一下即可。但实际代码中并不需要这个\(DP\)数组,可以直接求值。
#include <bits/stdc++.h>
typedef int mainint;
#define int long long
const int max_n=500000+4;
int N,Ans,root,tot;
int dis[max_n],first_edge[max_n];
struct Edge
{
int to,w,next_edge;
}edge[max_n<<1];
inline int read()
{
register int x=0,v=1;
register char ch=getchar();
while(!isdigit(ch))
{
if(ch=='-') v=-1;
ch=getchar();
}
while(isdigit(ch))
{
x=(x<<1)+(x<<3)+ch-'0';
ch=getchar();
}
return x*v;
}
inline void add_edge(int x,int y,int z)
{
edge[++tot].to=y;
edge[tot].next_edge=first_edge[x];
edge[tot].w=z;
first_edge[x]=tot;
}
void dfs1(int x,int fa)
{
for(int k=first_edge[x];k;k=edge[k].next_edge)
{
if(edge[k].to!=fa)
{
dfs1(edge[k].to,x);
dis[x]=std::max(dis[x],dis[edge[k].to]+edge[k].w);
}
}
}
void dfs2(int x,int fa)
{
for(int k=first_edge[x];k;k=edge[k].next_edge)
{
if(edge[k].to!=fa)
{
dfs2(edge[k].to,x);
Ans+=dis[x]-dis[edge[k].to]-edge[k].w;
}
}
}
mainint main()
{
int x,y,z;
N=read(),root=read();
for(int i=1;i<N;++i)
{
x=read(),y=read(),z=read();
add_edge(x,y,z),add_edge(y,x,z);
}
dfs1(root,0),dfs2(root,0);
printf("%lld\n",Ans);
return 0;
}