看到题目描述比较长的题目就想弃掉了……还要加强读题的能力啊。
题目大意:给定一棵无根树,取一条长度小于给定长度的树链,使得距离这条树链距离最远的节点最近,问最近距离。
首先我们知道以下两个性质:
- 对于树中的任意一点,距离其最远的点一定是树的直径的某一端点。
- 所有树的直径是等价的,即任意一条树的直径的最小偏心距相等。
于是我们可以先两次DFS求出树的直径,然后考虑以下两种情况:
- 当前节点在树的直径上,我们可以用一个类似于滑动窗口的东西来求出树的直径上的任意一点到任意一条树链的最小偏心距。
- 当前节点不在树的直径上,也就是当前节点的最小偏心距为当前节点到离该节点最近的树的直径上的节点的距离。
最后 O(n) 扫一遍所有节点求出最小偏心距即可。
附上AC代码:
#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;
const int N=5e5+10;
struct side{
int to,w,nt;
}s[N<<1];
int n,m,x,y,w,num,h[N],f[N],dis[N],l,r,ans;
bool b[N];
inline char nc(){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
return (void)(a*=f);
}
inline void add(int x,int y,int w){
s[++num]=(side){y,w,h[x]},h[x]=num;
s[++num]=(side){x,w,h[y]},h[y]=num;
}
inline void so(int x,int fa){
f[x]=fa;
for (int i=h[x]; i; i=s[i].nt)
if (!b[s[i].to]&&s[i].to!=fa) dis[s[i].to]=dis[x]+s[i].w,so(s[i].to,x);
return;
}
int main(void){
read(n),read(m);
for (int i=1; i<n; ++i) read(x),read(y),read(w),add(x,y,w);
r=1,so(1,0);
for (int i=1; i<=n; ++i) if (dis[i]>dis[r]) r=i;
dis[l=r]=0,so(r,0);
for (int i=1; i<=n; ++i) if (dis[i]>dis[l]) l=i;
ans=2e9;
for (int i=l,j=l; j!=r; i=f[i]){
while (f[j]&&dis[i]-dis[f[j]]<=m) j=f[j];
ans=min(ans,max(dis[l]-dis[i],dis[j]));
}
for (int i=l; i!=r; i=f[i]) b[i]=1;
for (int i=l; i!=r; i=f[i]) dis[i]=0,so(i,f[i]);
for (int i=1; i<=n; ++i) ans=max(ans,dis[i]);
printf("%d\n",ans);
return 0;
}