【LGP4886 】快递员

题目

好秒啊，真是一道神仙的点分治

于是我们来一个暴力的\(O(nlog^2n)\)的暴力统计吧

考虑计算每一个点作为快递中心时的答案

我们考虑在这个点成为分治重心时计算这个贡献

把这个贡献分成两部分

1. 分治块内部的点对，且不跨过分治重心，这个我们直接暴力统计就好了

2. 分治块外部，这个又分成两种，一种是跨过分治重心的路径，显然这样的点对的贡献就是路径长度；一种是两个点都在分治块外部

都在分治块外部的话好像很难去计算了，但是我们考虑尽管对于当前的分治重心都在分治块外部，但是对于这个点在点分树上的祖先，这种点对必然在某一个祖先的分治块内部，而且是在同一棵子树里的那种

于是我们对于每一个点暴力访问其在点分树上的祖先，算一下两点之间的距离

由于只需要计算最大的贡献，我们对于每一点都只保留最大的来自于同一子树的点对贡献

但是这样还是不科学，因为我们递归进这个子树的时候，最大值也可能来自这个子树，所以我们还需要保留次大值，就是为了在处理贡献最大子树的时候不出错

显然点分树的树高是\(logn\)的，但是我们需要计算树上距离需要一个链剖求\(lca\)，复杂度是\(O(nlog^2n)\)的

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=1e5+5;
const int inf=1e9;
inline int read() {
	char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
struct E{int v,nxt,w;}e[maxn<<1];
int vis[maxn],head[maxn],sum[maxn],top[maxn],son[maxn],fa[maxn],pre[maxn];
int d[maxn],Fa[maxn],h[maxn],mx[maxn],deep[maxn],col[maxn];
int S,now,rt,n,num,tmp,g,ans,m,ml;
int r[maxn<<1],len[maxn<<1];
std::vector<int> v[maxn],tp[maxn];
inline void add(int x,int y,int w) {
	e[++num].v=y;e[num].nxt=head[x];
	head[x]=num;e[num].w=w;
}
void getroot(int x,int fa) {
	sum[x]=1,mx[x]=0;
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[e[i].v]||e[i].v==fa) continue;
		getroot(e[i].v,x);sum[x]+=sum[e[i].v];
		mx[x]=max(mx[x],sum[e[i].v]);
	}
	mx[x]=max(mx[x],S-sum[x]);
	if(mx[x]<now) now=mx[x],rt=x;
}
void dfs1(int x) {
	sum[x]=1;int maxx=-1;
	for(re int i=head[x];i;i=e[i].nxt) {
		if(deep[e[i].v]) continue;
		deep[e[i].v]=deep[x]+1;fa[e[i].v]=x;
		pre[e[i].v]=pre[x]+e[i].w;
		dfs1(e[i].v);sum[x]+=sum[e[i].v];
		if(sum[e[i].v]>maxx) maxx=sum[e[i].v],son[x]=e[i].v;
	}
}
void dfs2(int x,int topf) {
	top[x]=topf;
	if(!son[x]) return;
	dfs2(son[x],topf);
	for(re int i=head[x];i;i=e[i].nxt)
	if(!top[e[i].v]) dfs2(e[i].v,e[i].v);
}
inline int LCA(int x,int y) {
	while(top[x]!=top[y]) {
		if(deep[top[x]]<deep[top[y]]) std::swap(x,y);
		x=fa[top[x]];
	}
	if(deep[x]<deep[y]) return x;return y;
}
inline int dis(int x,int y) {return pre[x]+pre[y]-2*pre[LCA(x,y)];}
void getro(int x,int fa,int c) {
	col[x]=c;
	for(re int j=0;j<v[x].size();j++) {
		int t=v[x][j];
		if(col[r[t^1]]==c) tmp=max(tmp,d[x]+d[r[t^1]]);
	}
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[e[i].v]||e[i].v==fa) continue;
		d[e[i].v]=d[x]+e[i].w;
		getro(e[i].v,x,c);
	}
}
void Dfs(int x,int fa) {
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[e[i].v]||e[i].v==fa) continue;
		Dfs(e[i].v,x);
	}
	col[x]=0;
}
void dfs(int x) {
	vis[x]=1;int cnt=0,m1=0,m2=0;
	g=ml;d[x]=0;
	for(re int i=head[x];i;i=e[i].nxt) {
		if(vis[e[i].v]) continue;
		col[x]=++cnt;tmp=0;d[e[i].v]=e[i].w;
		getro(e[i].v,0,cnt);
		tp[x].push_back(tmp);
		if(tmp>m1) m2=m1,m1=tmp;
			else m2=max(m2,tmp);
	}
	Dfs(x,0);int y=x;
	while(Fa[y]) {
		y=Fa[y];
		int mid=2*dis(x,y)+h[y];
		if(h[y]) g=max(g,mid);
	}
	g=max(g,m1);ans=min(ans,g);
	for(re int j=0,i=head[x];i;i=e[i].nxt,j++) {
		if(vis[e[i].v]) continue;
		S=sum[e[i].v],now=inf,getroot(e[i].v,0);
		if(tp[x][j]==m1) h[x]=m2;
			else h[x]=m1;
		Fa[rt]=x;dfs(rt);
	}
}
int main() {
	n=read(),m=read();ans=inf;
	for(re int x,y,z,i=1;i<n;i++)
		x=read(),y=read(),z=read(),add(x,y,z),add(y,x,z);
	deep[1]=1,dfs1(1),dfs2(1,1);
	int tot=1;
	for(re int i=1;i<=m;i++) {
		r[++tot]=read();
		v[r[tot]].push_back(tot);
		r[++tot]=read();
		v[r[tot]].push_back(tot);
		len[tot]=len[tot-1]=dis(r[tot],r[tot-1]);
		ml=max(ml,len[tot]);
	}
	S=n,now=inf,getroot(1,0);
	dfs(rt);
	printf("%d\n",ans);
	return 0;
}