树网的核加强版 (树的直径 尺取法 思维)

树网的核加强版 - 洛谷

比较明显的就是求出直径，这样路径最远的即为端点，比较好求；

那么我们求出直径后可以尺取直径，统计答案。

最后还有一种情况：路径上的一个分支的偏心距大于答案，这是完全可能的。

那么我们再对直径上的点dfs一次，再统计一次答案即可。

/*keep on going and never give up*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ll long long
#define db(x) cerr<<(#x)<<" "<<(x)<<" "<<endl;
#define endl "\n"
#define fast std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const double E = exp(1);
const double PI = acos(-1.0);
const int mod=1e9+7; 
const int maxn=5e5+10;
int n,m,s;
struct node{
	int to,w,nxt;
}e[maxn<<1];int tot,head[maxn<<1];
void add(int x,int y,int z){ e[++tot].to=y;e[tot].w=z;e[tot].nxt=head[x];head[x]=tot;}
int vis[maxn],dis[maxn],fa[maxn];
void dfs(int x){
	for(int i=head[x];i;i=e[i].nxt){
		int v=e[i].to,w=e[i].w;
		if(v==fa[x]||vis[v]) continue;
		fa[v]=x;
		dis[v]=dis[x]+w;
		dfs(v);
	}
}
signed main(){
	fast
	cin>>n>>s;
	int x,y,z;
	for(int i=1;i<n;i++){
		cin>>x>>y>>z;
		add(x,y,z);add(y,x,z);
	}
	int be=1,en=1;
	dis[be]=0;
	dfs(be);
	memset(fa,0,sizeof(fa));
	for(int i=1;i<=n;i++) if(dis[i]>dis[be]) be=i;
	dis[be]=0;
	dfs(be);
	for(int i=1;i<=n;i++) if(dis[i]>dis[en]) en=i;
	int ans=1e14,j=en;
	for(int i=en;i;i=fa[i]){
		while(fa[j]&&dis[i]-dis[fa[j]]<=s) j=fa[j]; //j->be
		ans=min(ans,max(dis[j],dis[en]-dis[i]));
	}
	for(int i=en;i;i=fa[i]) vis[i]=1;
	for(int i=en;i;i=fa[i]) dis[i]=0,dfs(i);
	for(int i=1;i<=n;i++) ans=max(ans,dis[i]);
	cout<<ans;
}