NOIP2018提高组D1T3 P5021 赛道修建

$Hyperlink$

https://www.luogu.org/problem/P5021

---

$Description$

现在有一棵大小为$n$的无根树，现在要求从中选择若干的道路（不能重复，可以不选完）使得这些道路中最短的那条最长，求这个最长长度

数据范围：$n\le 10^5$

---

$Solution$

首先肯定是要二分的（最小的最大嘛）

我们就二分这个最大长度$k$，问题就转换成为树中有多少条不重复的路径长度不小于$k$。

对于现在已经求出的一条路径值$va{l}_i$，如果已经大于$k$，那么直接$ans+1$，否则我们则用较小的匹配较大的，让其不小于$k$的同时尽量小，我们可以对于所有的$va{l}_{son}$，将其放入一棵平衡树中，这样就可以快速配对了

最后总的时间复杂度是：$O(nlognlogd)$，$d$是这张图的直径，上限为$\sum w_i$

$Tips$:二分的上界是树的直径，可以用两遍$dfs$/$bfs$或者树形$dp$求得，本人用的是$dp$。当然你可以直接设上界为$\sum w_i$，慢点罢了

据说还有$O(nlogd)$的做法？反正我太菜了不会。

---

$Code$

#include<set>
#include<cstdio>
#include<cctype>
#include<algorithm>
#define ri register int
using namespace std;int n,m,l[50001],p,q,tot,x,y,w;
struct node{int next,to,w;}e[100001];
inline void add(ri u,ri v,ri w){e[++tot]=(node){l[u],v,w};l[u]=tot;return;}
long long Ans,ans,f[50001],L=1,R;
multiset<int>s[50001];
multiset<int>::iterator it;
inline long long read()
{
	char c;int d=1;long long f=0;
	while(c=getchar(),!isdigit(c))if(c==45)d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
inline void dp(ri x,ri fa)
{
	for(ri i=l[x];i;i=e[i].next)
	{
		int y=e[i].to;
		if(y==fa) continue;
		dp(y,x);
		R=max(f[x]+f[y]+e[i].w,R);
		f[x]=max(f[x],f[y]+e[i].w);
	}
	return;
}
inline int dfs(int u,int fa,int k)
{
	s[u].clear();
	int val;
	for(register int i=l[u];i;i=e[i].next)
	{
		int v=e[i].to;
		if(v==fa) continue;
		val=dfs(v,u,k)+e[i].w;
		if(val>=k) ans++;
		else s[u].insert(val);
	}
	int now=0;
	while(s[u].size())
	{
		if(s[u].size()==1) return max(now,*s[u].begin());
		it=s[u].lower_bound(k-*s[u].begin());
		if(it==s[u].begin()&&s[u].count(*it)==1) it++;
		if(it==s[u].end()) {now=max(now,*s[u].begin());s[u].erase(s[u].find(*s[u].begin()));}
		else
		{
			ans++;
			s[u].erase(s[u].find(*it));
			s[u].erase(s[u].find(*s[u].begin()));
		}
	}
	return now;
}
bool check(int k)
{
	 ans=0;dfs(1,0,k);
	 return ans>=m;
}
signed main()
{
	n=read();m=read();
	for(register int i=1;i<n;i++)
	{
		x=read();y=read();w=read();
		add(x,y,w);add(y,x,w);
	}
	dp(1,0);
	while(L<=R)
	{
		int mid=L+R>>1;
		if(check(mid)) L=(Ans=mid)+1;
		else R=mid-1;
	}
	printf("%lld",Ans);
}