LCA 求公共父亲节点

DFS

//DFS
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<iomanip>
#define ll long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
#define RF(i,a,n) for(register int i=a;i>=n;i--)
#define NIL -1
#define MAX 100005
using namespace std;
struct node{
	int t,nex;
}e[100001];
int depht[50001], lg[50001], head[50001], father[50001][22];
int tot;
inline void add(int x,int y)
{
	e[++tot].t=y;
	e[tot].nex=head[x];
	head[x]=tot;
}
inline void dfs(int now,int fath)
{
	depht[now]=depht[fath]+1;
	father[now][0]=fath;
	for(register int i=1;(1<<i)<=depht[now];++i)
	  father[now][i]=father[father[now][i-1]][i-1];
	for(register int i=head[now];i;i=e[i].nex)
	{
		if(e[i].t!=fath)dfs(e[i].t,now);	
	}	
}
inline int lca(int x,int y)
{
	if(depht[x]<depht[y])
	  swap(x,y);
	while(depht[x]>depht[y])
	  x=father[x][lg[depht[x]-depht[y]]-1];
	if(x==y)
	  return x;
	for(register int k=lg[depht[x]];k>=0;--k)
	  if(father[x][k]!=father[y][k])
	    x=father[x][k],y=father[y][k];
	return father[x][0];
}
int n,m,s;
int main()
{
	//freopen("1.txt","r",stdin);
	scanf("%d%d%d",&n,&m,&s);
	for(register int i=1;i<=n-1;++i)
	{
		int x,y;scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs(s,0);
	for(register int i=1;i<=n;++i)
	  lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	for(register int i=1;i<=m;++i)
	{
		int x,y;scanf("%d%d",&x,&y);
		printf("%d\n",lca(x,y));
	}
	return 0;
}

BFS

//BFS
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<iomanip>
#include<queue>
#define ll long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
#define RF(i,a,n) for(register int i=a;i>=n;i--)
#define NIL -1
#define maxn 100005
using namespace std;
struct node{
	int to,d,next;
}a[maxn];
int pre[maxn],cnt=0;
int vis[maxn];
int dis[maxn],f[maxn][22],depht[maxn];
//len记录根节点到该点的长度,f记录向上的祖先节点,d记录每个节点的深度 
int T;

void add(int x,int y,int z){//链式前向星--加边 
	a[cnt].to=y; a[cnt].d=z; a[cnt].next=pre[x];
	pre[x]=cnt++;
}

void bfs(int u) {
	queue<int> q;
	memset(vis,0,sizeof(vis));
	memset(f,0,sizeof(f));
	q.push(u);
	vis[u]=1;//标记该点已经访问 
	depht[u]=1;//默认根节点深度为1,注意根节点深度不能为0 
	while(q.size())
	{
		int x=q.front();
		q.pop();
		for(int i=pre[x];~i;i=a[i].next){
			int y=a[i].to;
			if(!vis[y])
			{
				dis[y]=dis[x]+a[i].d;//和根节点的长度 
				depht[y]=depht[x]+1;//深度 
				vis[y]=1;//标记该点 
				f[y][0]=x;//存父节点 
				for(int j=1;j<=T;j++){
					f[y][j]=f[f[y][j-1]][j-1];//存储第f(y,2^j)个祖先 
				}
				q.push(y);
			}
		}		
	}
}

int LCA(int x,int y)
{
	if(depht[x] > depht[y])	swap(x,y);//保持 y>=x 
	for(int i = T; i >= 0; i--)//让y上升到与x同一深度 
		if(depht[f[y][i]] >= depht[x])	y = f[y][i];
	if(x == y)	return x;//如果此时相等,那就返回 
	for(int i = T; i >= 0; i--)//让x和y同时上升 同一段距离 
		if(f[x][i] != f[y][i])	x = f[x][i], y = f[y][i];
	return f[x][0];//x和y一定只差一步就相汇了,返回父节点。 
}	

int main(){
	ios::sync_with_stdio(false); cin.tie(0);
	int t,n,m;
	cin>>t;
	while(t--){
		cin>>n>>m;
		T = 22;
		//T=int(log(n)/log(2))+1;
		memset(pre,-1,sizeof(pre));
		cnt=0;
		for(int i=0;i<n-1;i++){//存边 
			int x,y,s;
			cin>>x>>y>>s;
			add(x,y,s);
			add(y,x,s);
		}
		bfs(1);
		while(m--){//q次询问 
			int x,y;
			cin>>x>>y;
			cout<<(dis[x]+dis[y]-2*dis[LCA(x,y)])<<endl;
		}
	}
	return 0;
 } 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值