luogu P1099 树网的核
题面传送门
先两遍\(dfs\)跑出树的直径。
因为所有树的直径都是相类似的,所以只要跑一条直径就好了。
然后在这条直径上尺取。
最后对于直径上每个点的子树不经过直径上的边求距离取最大值即可,这一步很难想,因为如果不取最大值那么就会漏掉答案因为无论如何都是要有这条边的权值的。
完美\(O(n)\)
代码实现:
#include<cstdio>
#include<cstring>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,x,y,z,r,now,cur,pus,maxn,d[1000039],ans=1e9,fa[1000039],flag[1000039];
struct yyy{int to,w,z;};
struct ljb{
int head,h[1000039];
yyy f[2000039];
inline void add(int x,int y,int z){
f[++head]=(yyy){y,z,h[x]};
h[x]=head;
}
}s;
inline void dfs(int x,int last){
fa[x]=last;
if(d[x]>d[maxn]) maxn=x;
int cur=s.h[x];
yyy tmp;
while(cur!=-1){
tmp=s.f[cur];
if(tmp.to!=last&&!flag[tmp.to]) d[tmp.to]=d[x]+tmp.w,dfs(tmp.to,x);
cur=tmp.z;
}
}
int main(){
register int i;
scanf("%d%d",&n,&m);
memset(s.h,-1,sizeof(s.h));
for(i=1;i<n;i++) scanf("%d%d%d",&x,&y,&z),s.add(x,y,z),s.add(y,x,z);
d[maxn]=0;dfs(1,0);
d[maxn]=0;dfs(maxn,0);
now=cur=r=maxn;
while(cur){
while(d[r]-d[cur]>m) r=fa[r];
ans=min(ans,max(d[maxn]-d[r],d[cur]));
flag[cur]=1;
cur=fa[cur];
}
for(i=1;i<=n;i++){
if(flag[i]){
maxn=i;d[i]=0;
dfs(maxn,0);
}
}
for(i=1;i<=n;i++) ans=max(ans,d[i]);
printf("%d\n",ans);
}