把博客园图标替换成自己的图标
把博客园图标替换成自己的图标end

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);
}
posted @ 2020-07-14 21:51  275307894a  阅读(26)  评论(0编辑  收藏  举报
浏览器标题切换
浏览器标题切换end