hdu2586 How far away? LCA

最新推荐文章于 2022-11-30 17:20:55 发布

原创最新推荐文章于 2022-11-30 17:20:55 发布 · 1.9k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

acm 同时被 3 个专栏收录

111 篇文章

订阅专栏

hdu

75 篇文章

订阅专栏

图论~

5 篇文章

订阅专栏

本文介绍了一种解决LCA（最近公共祖先）问题的方法，并通过HDU2586题目进行实现。文章详细展示了使用并查集与DFS遍历相结合的方式求解节点间的最近公共祖先及路径长度。

hdu2586

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define MAX 40000

struct edge{
	int v,w;
};
vector<edge> mp[MAX];
vector<edge> query[MAX];
bool flag[MAX];
int pre[MAX],father[MAX],path[MAX];

int find(int x){
	return x==pre[x]?	x:pre[x]=find(pre[x]);
}

void LCA(int k)
{
	int i,j;
	
	for(i=0;i<mp[k].size();i++)
	{
		int a=mp[k][i].v;
		if(!flag[a])
		{
			flag[a]=1;
			path[a]=path[k]+mp[k][i].w;
			LCA(a);
			pre[a]=k;
			for(j=0;j<query[a].size();j++)
			{
				int b=query[a][j].v;
				if(flag[b]&&father[query[a][j].w]==-1)
				{
					if(a==b)	father[query[a][j].w]=0;
					else	father[query[a][j].w]=path[a]+path[b]-2*path[find(b)];
				}
			}
		}
	}
}


int main()
{
	int i,j,k,T,n,m,a,b;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&m);
		
		for(i=1;i<=n;i++){
			mp[i].clear();
			query[i].clear();
			flag[i]=0;
			father[i]=-1;
			pre[i]=i;
			path[i]=0;
		}
		
		int a,b,c;
		edge X;
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			X.v=b;X.w=c;
			mp[a].push_back(X);
			X.v=a;
			mp[b].push_back(X);
		}
		for(i=1;i<=m;i++)
		{
			scanf("%d%d",&a,&b);
			X.v=b;X.w=i;
			query[a].push_back(X);
			X.v=a;
			query[b].push_back(X);
		}
		flag[1]=1;
	//	path[1]=0;		
		LCA(1);
		for(i=1;i<=m;i++)
			printf("%d\n",father[i]);
	}
	return 0;
}