POJ1470 - Closest Common Ancestors - (LCA tarjan)

POJ1470 - Closest Common Ancestors

• 链接 Closest Common Ancestors
• 贴一个Tarjan算法的讲解：优快云 - creatorx的博客

离线Tarjan :
利用并查集的合并。查询两个结点的公共祖先

先遍历树的 子节点 再将 根节点合并到父节点 ，其实就是理解 合并是一个树一个树进行的
合并遍历字节点的过程就是 当前以这个根节点为分叉点 找子节点，当要查询的两个子节点都遍历到时 共同祖先就是这个根节点

#include <iostream>
#include <algorithm>
#include <cstring>
#include <math.h>
#include <stdio.h>
#include <set>
#include <vector>
using namespace std;

const int N = 1e3+10;
int in[N], fa[N], vis[N], ask[N][N], n, num[N];
vector<int>g[N];

void init(){
    memset(in,0,sizeof in);
    memset(ask,0,sizeof ask);
    memset(vis, 0, sizeof vis);
    memset(num, 0 ,sizeof num);
    for(int i = 1; i <= n; i++){
        g[i].clear();
        fa[i] = i;
    }
}

int fin(int x){
    if(fa[x] == x) return x;
    return fa[x] = fin(fa[x]);
}

void lca(int r){
    for(int i = 1; i <= n; i++)
        if(vis[i] && ask[r][i]) num[fin(i)] += ask[r][i];
    
    vis[r] = 1;
    for(int i = 0; i < g[r].size(); i++){
        int t = g[r][i];
        lca(t);
        fa[t] = r;
    }
}

int main(){

    while(~scanf("%d",&n)){
        
        init();
        int x, m, y;

        for(int i = 1; i <= n; i++) {
            scanf("%d:(%d)",&x,&m);
            while(m--){
                scanf("%d",&y);
                g[x].push_back(y);
                in[y] ++;
            }
        }

        scanf("%d",&m);
        while(m--){
            scanf(" (%d %d)",&x,&y);
            ask[x][y] ++, ask[y][x] ++;
        }

        for(int i = 1; i <= n; i++){
            if(!in[i]) lca(i);
        }

        for(int i = 1; i <= n; i++) 
            if(num[i]) cout<<i<<':'<<num[i]<<endl;
    }
    return 0;

}