首先是最近公共祖先的概念(什么是最近公共祖先?):
在一棵没有环的树上,每个节点肯定有其父亲节点和祖先节点,而最近公共祖先,就是两个节点在这棵树上深度最大的公共的祖先节点。换句话说,就是两个点在这棵树上距离最近的公共祖先节点。他本身或者其父亲节点也可以作为祖先节点。
举个栗子吧,如下图所示4和5的最近公共祖先是2,5和3的最近公共祖先是1,2和1的最近公共祖先是1。
这就是最近公共祖先的基本概念了,那么我们该如何去求这个最近公共祖先呢?
通常初学者都会想到最简单粗暴的一个办法:对于每个询问,遍历所有的点,时间复杂度为O(n*q),很明显,n和q一般不会很小。 如:https://blog.csdn.net/henucm/article/details/82500028
常用的求LCA的算法有:Tarjan/DFS+ST/倍增。后两个算法都是在线算法,也很相似,时间复杂度在O(logn)~O(nlogn)之间,小弟能力有限 没看懂 。
这篇博客主要是介绍一下Tarjan算法。就是在一次遍历中把所有询问一次性解决所以其时间复杂度是O(n+q)。
Tarjan算法的优点在于相对稳定,时间复杂度也比较居中,也很容易理解。
下面是基本思路!!!
1.任选一个点为根节点,从根节点开始。
2.遍历该点u所有子节点v,并标记这些子节点v已被访问过。
3.若是v还有子节点,返回2,否则下一步。
4.合并v到u上。
5.寻找与当前点u有询问关系的点v。
6.若是v已经被访问过了,则可以确认u和v的最近公共祖先为v被合并到的父亲节点a。
遍历的话需要用到dfs来遍历,然后利用并查集来合并两个节点。
还是有点迷?? don't worry ! 客官你要的伪代码来啦~~
void Tarjan(u)// marge和find为并查集合并函数和查找函数
{
vis[u]=1;
for each(u,v) //访问所有u的子节点
{
if(!vis[i])// 如果这个子节点没有被访问
{
Tarjan(v);//继续往下遍历
link(u,v);// 合并v到u上
}
}
for each(u,e) //访问所有和u有访问关系的e
{
if(vis[i])//如果e被访问过
{
u.lca=v.lac=find(e) //u,e的最近公共祖先为find(e)
}
}
}
模拟一边给大家看 建议拿着纸和笔跟着我的描述一起模拟!!
假设我们有一组数据 9个节点 8条边 联通情况如下:
1--2,1--3,2--4,2--5,3--6,5--7,5--8,7--9 即下图所示的树
设我们要查找最近公共祖先的点为9--8,4--6,7--5,5--3;
设f[]数组为并查集的父亲节点数组,初始化f[i]=i,vis[]数组为是否访问过的数组,初始为0;
下面开始模拟过程:
取1为根节点,往下搜索发现有两个儿子2和3;
先搜2,发现2有两个儿子4和5,先搜索4,发现4没有子节点,则寻找与其有关系的点;
发现6与4有关系,但是vis[6]=0,即6还没被搜过,所以不操作;
发现没有和4有询问关系的点了,返回此前一次搜索,更新vis[4]=1;
表示4已经被搜完,更新f[4]=2,继续搜5,发现5有两个儿子7和8;
先搜7,发现7有一个子节点9,搜索9,发现没有子节点,寻找与其有关系的点;
发现8和9有关系,但是vis[8]=0,即8没被搜到过,所以不操作;
发现没有和9有询问关系的点了,返回此前一次搜索,更新vis[9]=1;
表示9已经被搜完,更新f[9]=7,发现7没有没被搜过的子节点了,寻找与其有关系的点;
发现5和7有关系,但是vis[5]=0,所以不操作;
发现没有和7有关系的点了,返回此前一次搜索,更新vis[7]=1;
表示7已经被搜完,更新f[7]=5,继续搜8,发现8没有子节点,则寻找与其有关系的点;
发现9与8有关系,此时vis[9]=1,则他们的最近公共祖先为find(9)=5;
(find(9)的顺序为f[9]=7-->f[7]=5-->f[5]=5 return 5;)
发现没有与8有关系的点了,返回此前一次搜索,更新vis[8]=1;
表示8已经被搜完,更新f[8]=5,发现5没有没搜过的子节点了,寻找与其有关系的点;
发现7和5有关系,此时vis[7]=1,所以他们的最近公共祖先为find(7)=5;
(find(7)的顺序为f[7]=5-->f[5]=5 return 5;)
又发现5和3有关系,但是vis[3]=0,所以不操作,此时5的子节点全部搜完了;
返回此前一次搜索,更新vis[5]=1,表示5已经被搜完,更新f[5]=2;
发现2没有未被搜完的子节点,寻找与其有关系的点;
又发现没有和2有关系的点,则此前一次搜索,更新vis[2]=1;
表示2已经被搜完,更新f[2]=1,继续搜3,发现3有一个子节点6;
搜索6,发现6没有子节点,则寻找与6有关系的点,发现4和6有关系;
此时vis[4]=1,所以它们的最近公共祖先为find(4)=1;
(find(4)的顺序为f[4]=2-->f[2]=2-->f[1]=1 return 1;)
发现没有与6有关系的点了,返回此前一次搜索,更新vis[6]=1,表示6已经被搜完了;
更新f[6]=3,发现3没有没被搜过的子节点了,则寻找与3有关系的点;
发现5和3有关系,此时vis[5]=1,则它们的最近公共祖先为find(5)=1;
(find(5)的顺序为f[5]=2-->f[2]=1-->f[1]=1 return 1;)
发现没有和3有关系的点了,返回此前一次搜索,更新vis[3]=1;
更新f[3]=1,发现1没有被搜过的子节点也没有有关系的点,此时可以退出整个dfs了。
板子:
#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<map>
#include<vector>
#define N 100005
using namespace std;
int father[30010*2],vis[30010*2];
int head[30010*2],head1[30010*2],dis[30010*2];
int root[30010*2],ans[30010*2];
int z,zz;
int n,m;
struct ac
{
int u,v,next,lca,w,id;
}r[30010*2],rr[30010*2];
void init()
{
z=zz=0;
memset(vis,0,sizeof(vis));
memset(head,-1,sizeof(head));
memset(head1,-1,sizeof(head1));
memset(root,1,sizeof(root));
memset(dis,0,sizeof(dis));
memset(ans,0,sizeof(ans));
for(int i=1;i<=n;i++)
{
father[i]=i;
}
}
void add(int u,int v,int w)
{
r[z].u=u;
r[z].v=v;
r[z].w=w;
r[z].next=head[u];
head[u]=z++;
}
void add1(int u,int v,int id)
{
rr[zz].u=u;
rr[zz].v=v;
rr[zz].id=id;
rr[zz].lca=-1;
rr[zz].next=head1[u];
head1[u]=zz++;
}
int find(int w)
{
if(w==father[w])
return w;
return father[w]=find(father[w]);
}
void link(int x,int y)
{
x=find(x);
y=find(y);
if(x!=y)
father[y]=x;
}
void taxjan(int cur)
{
vis[cur]=1;
for(int i=head[cur];i+1;i=r[i].next)
{
int v=r[i].v;
int w=r[i].w;
if(!vis[v])
{
dis[v]=dis[cur]+w;
taxjan(v);
link(cur,v);
}
}
for(int i=head1[cur];i+1;i=rr[i].next)
{
int v=rr[i].v;
if(vis[v])
{
int z=find(v);
ans[rr[i].id]=dis[cur]-2*dis[z]+dis[v];
}
}
}
int main()
{
cin>>n;
init();
for(int i=1;i<n;i++)
{
int u,v,w;
cin>>u>>v>>w;
root[v]=0;
add(u,v,w);
add(v,u,w);
}
int m,x;
cin>>m;
int c,d;
for(int i=1;i<=m;i++){
cin>>c>>d;
add1(c,d,i);
add1(d,c,i);
}
taxjan(1);
int sum=0;
for(int i=1;i<=m;i++)
{
sum+=ans[i];
}
cout<<sum<<endl;
return 0;
}