poj 1330（LCA）

最新推荐文章于 2020-08-26 15:11:40 发布

最新推荐文章于 2020-08-26 15:11:40 发布 · 79 阅读

本文提供了一个LCA（最近公共祖先）算法的实现模板，通过使用并查集来处理树状结构中节点之间的关系，适用于解决寻找两个节点最近公共祖先的问题。代码实现了节点的查找、递归处理以及询问处理等功能。

基本题，留着模板

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
using namespace std;

vector<int>lz[10005],que[10005];
int vis[10005],nopa[10005];
int n,m,pa[10005],s,e;

int find(int x)          
{
    if(pa[x] != x)
        return pa[x] = find(pa[x]);
    else
        return x;
}

void LCA(int u)
{
    int i,j;
    for(i=0; i<lz[u].size(); i++)      //递归处理
    {
        LCA(lz[u][i]);
        pa[lz[u][i]] = u;
    }
    vis[u] = 1;
    for(i=0; i<que[u].size(); i++)     //询问处理
    {
        if(vis[que[u][i]])
        {
            cout << find(que[u][i]) << endl;
            return;
        }
    }
}

int main()
{
    int t,i,j,a,b;
    cin >> t;
    while(t--)
    {
        cin >> n;
        for(i=1; i<=n; i++)       //初始化
        {
            lz[i].clear();
            que[i].clear();
            vis[i] = 0;
            nopa[i] = 0;
            pa[i] = i;
        }
        for(i=1; i<n; i++)
        {
            scanf("%d%d",&a,&b);
            lz[a].push_back(b);      //算是单向边，a是b的父结点
            nopa[b] = 1;          
        }
        scanf("%d%d",&s,&e);         
        que[s].push_back(e);
        que[e].push_back(s);         //询问两点连起关系
        for(i=1; i<=n; i++)
        {
            if(!nopa[i]) //从总结点开始LCA
            {
                LCA(i);
            }
        }
    }
    return 0;
}