LCA—Tarjan算法

#include<bits/stdc++.h>
using namespace std;
const int SIZE =50010;
struct edge
{
    int next,y,v;
}e[SIZE*2];
int to[SIZE];
int fa[SIZE],d[SIZE],v[SIZE],lca[SIZE],ans[SIZE];
vector<int> query[SIZE],query_id[SIZE];
int T,n,m,tot,t,len=0;
void insert(int xx,int yy,int vv)
{
    e[++len].next=to[xx];
    to[xx]=len;
    e[len].y=yy;
    e[len].v=vv;
}
void add_query(int x,int y,int id)
{
    query[x].push_back(y),query_id[x].push_back(id);
    query[y].push_back(x),query_id[y].push_back(id);
}
int getf(int x)
{
    if(x==fa[x]) return x;
    return fa[x]==getf(fa[x]);
}
void tarjan(int x)
{
    v[x]=1;
    for(int i=to[x]; i;i=e[i].next)
    {
        int y=e[i].y;
        if(v[y]) continue;
        d[y] = d[x] + e[i].v;
        tarjan(y);
        fa[y]=x;
    }
    for(int i=0;i < query[x].size();i++)
    {
        int y=query[x][i],id=query_id[x][i];
        if(v[y]==2)
        {
            int lca = getf(y);
            ans[id]=min(ans[id],d[x]+d[y]-2*d[lca]);
        }
    }
    v[x]=2;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            to[i]=0; fa[i]=i; v[i]=0;
            query[i].clear(),query_id[i].clear();
        }
        tot = 0;
        for(int i=1;i<n;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            insert(x,y,z);
            insert(y,x,z);
        }
        for(int i=1;i<=m;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(x==y)    ans[i]=0;
            else{
                add_query(x,y,i);
                ans[i]= 1<<30;
            }
        }
        tarjan(1);
        for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    }
}

向上标记法

从x向上走到根节点，并标记所有经过的节点。

从y向上走到根节点，当第一次遇到已标记的节点，就找到了LCA(x,y).

对于每个询问，向上标记法的时间复杂度为O(n).m个询问时间复杂度则为O(nm).

Tarjan 算法

Tarjan是使用并查集对“向上标记法”的优化，需要将m个询问一次性输入，重新组织查询处理顺序，统一计算，最后统一输出。时间复杂度为O(n+m)。

思路：进入一个节点x的深搜，把整个树的一部分看作以节点x为根节点的小树，再搜索其他节点。每搜索完一个节点之后，如果该点和另一个已搜索完点为需要查询LCA的点，则这两点的LCA为另一个点现在的祖先。

回顾一波深度优先遍历（DFS）

关于树的DFS序

树的DFS序的特点就是对于每个节点恰好出现两次。

深度优先遍历，树中节点分三类：

已经访问完毕并且回溯的节点，在节点上标记一个整数2.
已经开始递归，但尚未回溯的节点。这些节点就是当前节点正在访问的节点x以及x的祖先。在这些节点上标记一个整数1.
对于尚未访问的节点。这些节点没有标记。

具体操作：

任选一个点为根节点，从根节点开始遍历该点x的所有子节点y，并标记已被访问。
如果y也有子节点，重复操作二，否则合并y到x上。
寻找与当前点x有询问关系的点y。因为是离线算法，所以将需要从询问找出与当前点有询问关系的点，即为离线做法的特点。
如果找出来的y已经被访问过了，则可以确认x和y的最近公共祖先为y在并查集中的代表元素。

注意：Tarjan算法是离线算法，若题目强制在线（如针对几种操作，每一次求LCA操作后都要求输出答案），这时候Tarjan就不再适用。

【HDU2586】

题意：将n个点用n-1条边相连，m次询问a，b两个点之间的距离。

分析：这就是一道LCA的板子题。明显可用离线算法Tarjan做。

猜你喜欢