`
xuela_net
  • 浏览: 495097 次
文章分类
社区版块
存档分类
最新评论

poj 1330(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;
}


分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics