HDU 2586-How far away(最近公共祖先LCA)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2586


大致题意:给你一棵树,树上有n个点,n-1条边,有q次查询,每次查询树上任意两点间的距离。


题解:LCA经典题,定义一个dis数组表示该点到树根(假如有根的话)的距离,则根据LCA中的

Tarjan算法的工作机制和原理,保存查询的两点,之后跑一发Tarjan,有关系(其中待求的两点)

的两点,假如另一点已经访问,则找到这两点的最近公共祖先,很容易推出,这两点间的距离就是

dfs[ i ]+dis[ j ]-2*dis[ z[i] ];其中z[i]为i和j的最近公共祖先。


拓展:最近公共祖先及Tarjan算法讲解


#include<map>  
#include<stack>  
#include<queue>  
#include<vector>  
#include<math.h>  
#include<stdio.h>  
#include<iostream>  
#include<string.h>  
#include<stdlib.h>  
#include<algorithm>  
using namespace std;  
typedef long long  ll;  
#define inf 1000000000  
#define mod 1000000007   
#define  maxn  40010  
#define  lowbit(x) (x&-x)  
#define  eps 1e-10  
int dis[maxn],vis[maxn],n,m,e1[maxn],e2[maxn],fa[maxn],e3[maxn];
//e1和e2数组分别存待求的两点,e3存这两点的最近公共祖先
struct node
{
	int id,c;
};
vector<node>q[maxn];
int find(int x)
{
	if(fa[x]==x)
		return x;
	return fa[x]=find(fa[x]);
}
void tarjan(int x)
{
	int i;
	vis[x]=1;fa[x]=x;
	for(i=1;i<=m;i++)
	{
		if(e1[i]==x && vis[e2[i]])
			e3[i]=find(e2[i]);
		if(e2[i]==x && vis[e1[i]])
			e3[i]=find(e1[i]);
	}
	for(i=0;i<q[x].size();i++)
	{
		node v=q[x][i];
		if(vis[v.id])
			continue;
		dis[v.id]=dis[x]+v.c;
		tarjan(v.id);
		fa[v.id]=x;
	}
}
int   main(void)
{
	node tmp;
	int T,i,x,y,z;
	scanf("%d",&T);
	while(T--)
	{
		memset(e1,0,sizeof(e1));
		memset(e2,0,sizeof(e2));
		memset(e3,0,sizeof(e3));
		memset(vis,0,sizeof(vis));
		memset(dis,0,sizeof(dis));
		scanf("%d%d",&n,&m);
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&z);
			tmp.id=x;tmp.c=z;
			q[y].push_back(tmp);
			tmp.id=y;tmp.c=z;
			q[x].push_back(tmp);
		}
		for(i=1;i<=m;i++)
			scanf("%d%d",&e1[i],&e2[i]);
		dis[1]=0;//默认1为树根
		tarjan(1);
		for(i=1;i<=m;i++)
			printf("%d\n",dis[e1[i]]+dis[e2[i]]-2*dis[e3[i]]);
		for(i=0;i<=maxn;i++)
			if(q[i].empty()==0)
				q[i].clear();
	}
	return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值