CD操作

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<iostream>
#include<string>
#include<map>
using namespace std;

typedef __int64 lld;
const int maxn=100005;
int n,m;
int dp[maxn][18];

vector<int>edge[maxn];
map<string,int>mp;
string s1,s2;
int tot;
int root;
int deep[maxn];

void dfs(int u,int f,int dep){
	int i;
	deep[u]=dep;
	dp[u][0]=f;
	for(i=1;i<18;i++){
		dp[u][i]=dp[dp[u][i-1]][i-1];
	}
	for(i=0;i<edge[u].size();i++){
		dfs(edge[u][i],u,dep+1);
	}
}

int Ancestor(int x,int len){
	for(int i=17;i>=0;i--){
		if(len>=1<<i){
			len-=1<<i;
			x=dp[x][i];
		}
	}
	return x;
}

int LCA(int x,int y){
	if(deep[x]>deep[y])
		x=Ancestor(x,deep[x]-deep[y]);
	else
		y=Ancestor(y,deep[y]-deep[x]);
	if(x==y)return x;
	int ret=root;
	int i=17;
	while(i--){
		if(dp[x][i]==dp[y][i]){
			ret=dp[x][i];
		}else{
			x=dp[x][i];
			y=dp[y][i];
		}
	}
	return ret;
}

int main(){
	int i,j;
	int T;
	int a,b;
	int u,v;
	scanf("%d",&T);
	while(T--){
		scanf("%d %d",&n,&m);
		tot=1;
		for(i=0;i<maxn;i++)
			edge[i].clear();
		mp.clear();
		for(i=1;i<n;i++){
			cin>>s1>>s2;
			if(mp.find(s1)==mp.end()){
				a=tot;
				mp[s1]=tot++;
			}else
				a=mp[s1];
			if(mp.find(s2)==mp.end()){
				b=tot;
				mp[s2]=tot++;
			}else
				b=mp[s2];
			edge[b].push_back(a);
			deep[a]=1;
		}
		
		for(i=1;i<tot;i++){
			if(!deep[i])root=i;
		}
		memset(dp,0,sizeof(dp));
		dfs(root,0,0);
		for(i=1;i<=m;i++){
			cin>>s1>>s2;
			a=mp[s1];
			b=mp[s2];
			int ff=LCA(a,b);
			int ans=deep[a]-deep[ff]+(b!=ff);
			printf("%d\n",ans);
		}
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值