[NOI2003] 逃学的小孩
题目描述
Chris 家的电话铃响起了，里面传出了 Chris 的老师焦急的声音：“喂，是 Chris 的家长吗？你们的孩子又没来上课，不想参加考试了吗？”一听说要考试，Chris 的父母就心急如焚，他们决定在尽量短的时间内找到 Chris。他们告诉 Chris 的老师：“根据以往的经验，Chris 现在必然躲在朋友 Shermie 或 Yashiro 家里偷玩《拳皇》游戏。现在，我们就从家出发去找 Chris，一旦找到，我们立刻给您打电话。”说完砰的一声把电话挂了。
Chris 居住的城市由 \(N\) 个居住点和若干条连接居住点的双向街道组成，经过街道 \(x\) 需花费 \(T_{x}\) 分钟。可以保证，任意两个居住点间有且仅有一条通路。Chris 家在点 \(C\)，Shermie 和 Yashiro 分别住在点 \(A\) 和点 \(B\)。Chris 的老师和 Chris 的父母都有城市地图，但 Chris 的父母知道点 \(A\)、\(B\)、\(C\) 的具体位置而 Chris 的老师不知。
为了尽快找到 Chris，Chris 的父母会遵守以下两条规则：
- 如果 \(A\) 距离 \(C\) 比 \(B\) 距离 \(C\) 近，那么 Chris 的父母先去 Shermie 家寻找 Chris，如果找不到，Chris 的父母再去 Yashiro 家；反之亦然。
- Chris 的父母总沿着两点间唯一的通路行走。
显然，Chris 的老师知道 Chris 的父母在寻找 Chris 的过程中会遵守以上两条规则，但由于他并不知道 \(A\)、\(B\)、\(C\) 的具体位置，所以现在他希望你告诉他，最坏情况下 Chris的父母要耗费多长时间才能找到 Chris？
输入格式
输入文件第一行是两个整数 \(N\) 和 \(M\)，分别表示居住点总数和街道总数。
以下 \(M\) 行，每行给出一条街道的信息。第 \(i+1\) 行包含整数 \(U_{i}\)、\(V_{i}\)、\(T_{i}\)，表示街道 \(i\) 连接居住点 \(U_{i}\) 和 \(V_{i}\)，并且经过街道 \(i\) 需花费 \(T_{i}\) 分钟。街道信息不会重复给出。
输出格式
输出文件仅包含整数 \(T\)，即最坏情况下 Chris 的父母需要花费 \(T\) 分钟才能找到 Chris。
样例 #1
样例输入 #1
4 3
1 2 1
2 3 1
3 4 1
样例输出 #1
4
提示
对于 \(100\%\) 的数据，\(3 \le N \le 2\times 10^5\)，\(1 \le U_{i},V_{i} \le N\)，\(1 \le T_{i} \le 10^{9}\)。
分析
可以发现原图是一棵树。
对于路径 $ C $ --> $ A $ --> $ B $ ，( $ len(A,C) $ < $ len(A,B) $ )
所以我们希望使 $ len(A,B) $ 尽可能大，即树的直径。
只需要使 $ len(A,C) $ 即到直径端点距离最大即可。
dfs找直径长和到直径端点的最大距离。
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+100;
typedef long long ll;
struct edge{int y,n;ll z;}e[N<<1];
int n,m;
int head[N],cnt,son[N];
int root1,root2,emp;
ll f[N][2];
ll dis[3][N];
void ad(int x,int y,ll z)
{
e[++cnt].n=head[x];
e[cnt].y=y;
e[cnt].z=z;
head[x]=cnt;
}
void init()
{
scanf("%d%d",&n,&m);
for(int i=1,x,y;i<n;++i)
{
ll z;
scanf("%d%d%lld",&x,&y,&z);
ad(x,y,z);
ad(y,x,z);
}
}
void go(int op,int u,int fa,int &root)
{
if(dis[op][u]>dis[op][root])
root=u;
for(int i=head[u];i;i=e[i].n)
{
int v=e[i].y;
if(v==fa)continue;
dis[op][v]=dis[op][u]+e[i].z;
go(op,v,u,root);
}
}
void work()
{
go(0,1,0,root1);
go(1,root1,0,root2);
go(2,root2,0,emp);
ll ans=0;
for(int i=1;i<=n;++i)
{
ll nw=min(dis[1][i],dis[2][i]);
ans=max(nw+dis[1][root2],ans);
}
cout<<ans;
}
int main()
{
init();
work();
return 0;
}