hdu 2586 How far away LCA在线RMQ(模板题)

点击打开链接

求dist(u,v) 用最短路flold显然超时 由于是树,则可以O(n)求出w=LCA(u,v)
dist(u,v)=d(u)+d(v)-2*d(w); d(i)为根到结点i的距离 

计算w=LCA（u，v） w为u->v路径中深度最小的点 则先用dfs记录u->v路径的遍历顺序,然后RMQ求出该路径中(pos[u]~pos[v])中dep最小的编号x 返回vs[x]即可

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include <vector>
using namespace std;
const int N=8e4+20;
typedef pair<int,int> ii;
typedef long long ll; 
int n,q,vis[N],num;
int dis[N];//根到结点的距离 
int dep[N];//第i次访问的结点深度
int vs[N];//vs[i] dfs第i次访问到的结点
int pos[N];//pos[v] 结点v第一次被访问到的位置 
vector<ii> e[N];
int d[N][30];//d[i][j] 序号i~2^j的最小dep序号 
void init()
{
	for(int i=0;i<=n;i++)
	e[i].clear();
	
	num=0;
	memset(dis,0,sizeof(dis));
	memset(vis,0,sizeof(vis));
	memset(pos,-1,sizeof(pos));
}
void dfs(int u,int depth)
{
	vs[++num]=u;
	dep[num]=depth;
	if(pos[u]==-1)
	pos[u]=num;
	
	vis[u]=1;
	for(int i=0;i<e[u].size();i++)
	{
		int v=e[u][i].first;
		int w=e[u][i].second;
		if(vis[v])
		continue;
		
		dis[v]=dis[u]+w;
		dfs(v,depth+1);
		
		//记录路径
		vs[++num]=u;
		dep[num]=depth;
		 
	}
}
void RMQ(int len)//求(u,v)路径中深度最小的结点的访问序号即LCA(u,v) 
{
	for(int i=0;i<=len;i++)
	d[i][0]=i;
	
	for(int j=1;(1<<j)<=len;j++)
	{
		for(int i=1;i+(1<<j)<=len;i++)
		{
			int x=d[i][j-1],y=d[i+(1<<(j-1))][j-1];
			if(dep[x]<=dep[y])
			d[i][j]=x;
			else
			d[i][j]=y;	
		} 
	}
}
int LCA(int x,int y)
{
	int a=pos[x],b=pos[y];
	if(a>b)
	swap(a,b);
	
	//访问序号(a,b)中深度最小的点即为LCA(x,y)
	int k=0;
	while((1<<k+1)<=b-a+1) k++;
	
	int l=d[a][k],r=d[b-(1<<k)+1][k];
	if(dep[l]<dep[r])
	return vs[l];
	else
	return vs[r]; 
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>q;
		init();
		for(int i=1;i<=n-1;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			e[u].push_back(ii(v,w));
			e[v].push_back(ii(u,w));
		}
		//求dist(u,v) 用最短路flold显然超时 由于是树,则可以O(n)求出w=LCA(u,v)
		//dist(u,v)=d(u)+d(v)-2*d(w); d(i)为根到结点i的距离 
		dfs(1,1);
		RMQ(num);
		while(q--)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			ll ans=dis[u]+dis[v]-2*dis[LCA(u,v)];
			cout<<ans<<endl;
		}
	} 
	return 0;
}