牛客练习赛67 F牛妹的苹果树 树上直径合并/ dfs序求LCA + 倍增维护最远点对

经典问题。

证明类似于树的直径。

两个点集所在的联通快合并，(其中可能会有交集)。其中合并后直径(点集中最远点对)为，两个联通快的直径对应的4个点，组合的6中情况的点对最大值。(即直径一定是这4个点中的2个，证明类似树的直径，从任一点出发，所达最远点是直径点，显然这4个点任一点出发，最远点一定还是这4个点中的一个)

然后用dfs序维护lca(可以做到O1)，倍增维护树直径合并即可

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M = 3e6+7;
int head[M],cnt=1;
struct EDGE{int to,nxt;ll w;}ee[M*2];
void add(int x,int y,ll w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}

int sx[M][21],sy[M][21];
int n;
ll dist[M];//点i到根节点的距离
int st[M],ed[M],dep[M],lg[M*2],dp[M*2][21];
int euler[M*2],sz=0;
void dfs(int u,int fa)
{
	euler[++sz]=u;
	st[u]=sz;
	dep[u]=dep[fa]+1;
	for(int i=head[u];i;i=ee[i].nxt)
	{
		int v=ee[i].to;
		if(fa==v)continue;
		dist[v]=dist[u]+ee[i].w;
		dfs(v,u);
		euler[++sz]=u;
	}
	ed[u]=sz;
}
int query(int l,int r)
{
	int k=lg[r-l+1];
	return min(dp[l][k],dp[r-(1<<k)+1][k]);
}
inline int lca(int u,int v)
{
	if(u==v)return u;
	int x=min(st[u],st[v]),y=max(ed[v],ed[u]);
	if(x>y)swap(x,y);
//	cout<<x<<" "<<y<<"  "<<euler[query(x,y)]<<"  "<<u<<" "<<v<<endl;
	return euler[query(x,y)];
}
inline ll dis(int x,int y)
{
	return dist[x]+dist[y]-dist[lca(x,y)]*2;
}
inline pair<int,int>cp(int a,int b,int c,int d)
{
	int x=a,y=b;
	if(dis(x,y)<=dis(a,b))x=a,y=b;
	if(dis(x,y)<=dis(a,c))x=a,y=c;
	if(dis(x,y)<=dis(a,d))x=a,y=d;
	if(dis(x,y)<=dis(b,c))x=b,y=c;
	if(dis(x,y)<=dis(b,d))x=b,y=d;
	if(dis(x,y)<=dis(c,d))x=c,y=d;
	return {x,y};
}

void gao()
{
	sz=0;
	dfs(1,0);
	for(int i=1;i<=sz;i++)
		dp[i][0]=st[euler[i]];
	for(int j=1;j<=20;j++)
		for(int i=1;i+(1<<j)-1<=sz;i++)
			dp[i][j]=min(dp[i][j-1],dp[i+(1<<j-1)][j-1]);
	
	for(int i=1;i<=n;i++)sx[i][0]=i,sy[i][0]=i;
	for(int i=1;i<=20;i++)
	for(int j=1;j + (1 << i) - 1 <= n;j++)
	{
		int a=sx[j][i-1],b=sy[j][i-1],c=sx[j+(1<<(i-1))][i-1],d=sy[j+(1<<(i-1))][i-1];
		pair<int,int> p = cp(a,b,c,d);
		sx[j][i]=p.first,sy[j][i]=p.second;
	}
}

int main()
{
	lg[0]=-1,lg[1]=0;for(int i=2;i<=6e5;i++)lg[i]=lg[i>>1]+1;
	int q,u,v,w;
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++)scanf("%d%d%d",&u,&v,&w),add(u,v,w),add(v,u,w);
	gao();
	while(q--)
	{
		int l,r;
		scanf("%d%d",&l,&r);
		int k=lg[r-l+1];
		int a,b,c,d;
		a=sx[l][k],b=sy[l][k];
		c=sx[r-(1<<k)+1][k],d=sy[r-(1<<k)+1][k];
		pair<int,int> p = cp(a,b,c,d);
		printf("%lld\n",dis(p.first,p.second));
	}
	return 0;
}