HDU 4547 CD操作 LCA

题意;n个结点的树,根到任意一个结点只要一秒,子结点往根走需要1秒.
m次询问,u->v的最小时间. n,m<=1e5.

u->v  = u->lca(u,v)->v 时间为u到lca(u,v)的距离加上1

用倍增处理出lca即可.O(nlogn).注意u-v相同的情况和n==1的情况.

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+20;
const int M=22;
map<string,int> mp;
vector<int> e[N];
int d[N],p[N][25];
void dfs(int u,int fa)
{
	if(fa)
		d[u]=d[fa]+1;
	p[u][0]=fa;
	for(int i=1;i<M;i++)
		p[u][i]=p[p[u][i-1]][i-1];
	for(int i=0;i<e[u].size();i++)
	{
		int v=e[u][i];
		if(v==fa)	continue;
		dfs(v,u);
	}
}
int lca( int a, int b ){  
    if( d[a] > d[b] ) a ^= b, b ^= a, a ^= b;  
    if( d[a] < d[b] )
	{  
        int del = d[b] - d[a];  
        for( int i = 0; i < M; i++ ) if(del&(1<<i)) b=p[b][i];  //让a,b深度相同.
    }  
    if( a != b )
	{  
        for( int i = M-1; i >= 0; i-- )   
            if( p[a][i] != p[b][i] )   //祖先一样上不跳 防止跳太上面.
                 a = p[a][i] , b = p[b][i];  
        a = p[a][0], b = p[b][0]; //此时在最逼近lca点,在跳一次就好了. 
    }  
    return a;  
}
int vis[N];
int main()
{
	//std::ios::sync_with_stdio(false);
	int T;
	scanf("%d",&T);
	while(T--)
	{
	
		int num=0,n,m;
		scanf("%d%d",&n,&m);
		for(int i=0;i<=n;i++)
			e[i].clear(),vis[i]=0,d[i]=0;
		mp.clear();
		string a,b;
		for(int i=1;i<=n-1;i++)
		{
			cin>>a>>b;
			if(!mp[a])
				mp[a]=++num;
			if(!mp[b])
				mp[b]=++num;
			int u=mp[a],v=mp[b];
			vis[u]=1;
			e[u].push_back(v),e[v].push_back(u);	
		}
		int rt;
		for(int i=1;i<=n;i++)
		{
			if(vis[i]==0)
			{
				rt=i;
				break;
			}
		}
		dfs(rt,0);
		while(m--)
		{
			cin>>a>>b;
			if(n==1||a==b)
			{
				printf("0\n");
				continue;
			}
			int u=mp[a],v=mp[b];
			int x=lca(u,v);
			int res=d[u]-d[x];
			if(v!=x)
				res++;
			printf("%d\n",res);
		}
	}
	return 0;
}