/*
http://acm.hdu.edu.cn/showproblem.php?pid=2586
题意:
一个村子里有n个房子,这n个房子用n-1条路连接起来,接下了有m次询问,每次询问两个房子a,b之间的距离是多少。
思路:
很明显的最近公共祖先问题,先建一棵树,然后求出每一点i到树根的距离dis[i],然后每次询问a,b之间的距离=dis[a]+dis[b]-2*dis[LCA(a,b)];
LCA(a,b)即是a,b的最近公共祖先。。
*/
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<vector>
using namespace std;
const int maxn = 40005;
int n;
int p[maxn];
int inq[maxn];
bool vis[maxn];
int dis[maxn],ans[maxn];
struct node
{
int id,w; //w在tree中为路的距离 query为提问的顺序
node( int a,int b ){ id = a; w = b; }
};
vector<node>tree[maxn],query[maxn]; //邻接表 询问
int find( int x ) //并查集
{
if( p[x] == x )
return x;
else
return p[x] = find(p[x]);
}
void merge( int a,int b ) //合并并查集
{
int x = find(a);
int y = find(b);
p[y] = x;
}
void LCA( int u )
{
vis[u] = true;
for( int i =0; i < tree[u].size(); i ++ )
{
int v = tree[u][i].id;
if( !vis[v] )
{
dis[v] = dis[u] + tree[u][i].w;
LCA( v );
merge( u,v ); //注意顺序, 先Tarjan子节点tmp, 再更新其father[tmp], 因为要保证在递推tmp所代表的子树时, father[tmp] = tmp, 而与当前子树无关. 递归回来的时候再把tmp代表的子树`并入`到当前树里
}
}
for( int i =0; i < query[u].size(); i ++ )
{
int v = query[u][i].id;
if( vis[v] )
{
ans[ query[u][i].w ] = dis[u] + dis[v] - 2*dis[find(v)];
}
}
}
void init() //初始化
{
for( int i = 1; i <= n ; i ++ )
{
p[i] = i;
tree[i].clear();
query[i].clear();
}
memset( inq,0,sizeof(inq) );
memset( vis,0,sizeof(vis) );
memset( dis,0,sizeof(dis) );
}
int main()
{
int t,m,a,b,c;
scanf("%d",&t);
while( t-- )
{
scanf("%d%d",&n,&m);
init();
for( int i = 1; i < n; i ++ )
{
scanf("%d%d%d",&a,&b,&c);
tree[a].push_back(node(b,c));
tree[b].push_back(node(a,c));
}
for( int i = 0; i < m; i ++)
{
scanf("%d%d",&a,&b);
query[a].push_back( node(b,i) );
query[b].push_back( node(a,i) );
}
LCA(1);
for( int i = 0; i < m; i ++ )
{
printf("%d\n",ans[i]);
}
}
return 0;
}
/*
/*
最近公共祖先lca 离线算法/Tarjan算法
方法举例说明:
1
/ \
2 3
/ \
4 5
/ /
7 8
/
9
查询(4,9):到4时,由于vis[9]=0,所以继续;到9后,最近公共祖先就是f[4]=4(回溯的时候记录父节点);
查询(9,8):深度优先搜索,所以到8时才询问;要到8必须回溯到2,在进到8这个子树,所以以记录f[9]=7;f[7]=4;f[4]=2;f[2]=2;所以find(2)=2;
查询(8,3):跟上条相似,必须回溯1才能到3,而find(8)=1就是最近公共祖先;
我们可以发现,对于查询的两点,都要在先查询到的点开始,回溯到最近公共祖先,才查询相对应的点。这也就是离线算法的思路了。。这是我自己理解的。。
下面是专业说明,反正我是没看懂。。
利用并查集优越的时空复杂度,我们可以实现LCA问题的O(n+Q)算法,这里Q表示询问的次数。
Tarjan算法基于深度优先搜索的框架,对于新搜索到 的一个结点,首先创建由这个结点构成的集合,
再对当前结点的每一个子树进行搜索,每搜索完一棵子树,则可确定子树内的LCA询问都已解决。
其他的LCA询问的结果必然在这个子树之外,这时把子树所形成的集合与当前结点的集合合并,并将当前结点设为这个集合的祖先。
之后继续搜索下一棵子树,直到当前结点的所 有子树搜索完。这时把当前结点也设为已被检查过的,
同时可以处理有关当前结点的LCA询问,如果有一个从当前结点到结点v的询问,
且v已被检查过,则由于 进行的是深度优先搜索,当前结点与v的最近公共祖先一定还没有被检查,
而这个最近公共祖先的包涵v的子树一定已经搜索过了,那么这个最近公共祖先一定是v 所在集合的祖先。
*/
*/
HDU - 2586 LCA
最新推荐文章于 2022-11-30 17:20:55 发布