求助: 倍增LCA20分
查看原帖
求助: 倍增LCA20分
535758
_mumu楼主2021/8/9 18:18

代码如下

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

vector <int> in[500001];
int dp[500001],f[500001][20];

void deepfind(int find,int deep,int fa)
{
	for(int i=0;i<in[find].size();i++)
	{
		if(in[find][i]!=fa)
		{
			dp[in[find][i]]=deep+1;
			f[in[find][i]][0]=find;
			deepfind(in[find][i],deep+1,find);
		}
	}
}

int LCA(int x,int y)
{
	for(int j=19;j>-1;j--)
	{
		if(dp[f[y][j]]>=dp[x])
		{
			y=f[y][j];
		}
	}
	
	if(x==y)
	{
		return x;
	}
	else
	{
		for(int j=19;j>-1;j--)
		{
			if(f[x][j]!=f[y][j])
			{
				x=f[x][j];
				y=f[y][j];
			}
		}
		return f[x][0];
	}
}

int main()
{
	int n,m,s;
	scanf("%d%d%d",&n,&m,&s);
	
	int a,b;
	for(int i=0;i<n-1;i++)
	{
		scanf("%d%d",&a,&b);
		in[a].push_back(b);
		in[b].push_back(a);
	}
	
	dp[s]=1;
	deepfind(s,1,0);
	
	for(int i=0;i<20;i++)
	{
		for(int j=1;j<n+1;j++)
		{
			f[i][j]=f[f[i][j-1]][j-1];
		}
	}
	
	for(int i=0;i<m;i++)
	{
		scanf("%d%d",&a,&b);
		int x=b,y=a;
		if(dp[a]<dp[b])
		{
			x=a;
			y=b;
		}
		printf("%d\n",LCA(x,y));
	}
}
2021/8/9 18:18
加载中...