码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 1330

时间:2016-05-02 15:29:56      阅读:122      评论:0      收藏:0      [点我收藏+]

标签:

题意给一个有根树,一个查询节点(u,v)的最近公共祖先;

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <algorithm>
#include <set>
using namespace std;
typedef long long ll;
typedef unsigned long long Ull;
#define MM(a,b) memset(a,b,sizeof(a));
const double eps = 1e-10;
const int  inf =0x7f7f7f7f;
const double pi=acos(-1);
const int maxn=10000;
int dep[maxn+10],par[maxn+10];
int n,a,b,u,v;
vector<int> G[maxn+10];

void dfs(int cur,int d)
{
    dep[cur]=d;
    for(int i=0;i<G[cur].size();i++)
         dfs(G[cur][i],d+1);
}

int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) G[i].clear();
        MM(par,0);
        for(int i=1;i<=n-1;i++)
        {
            scanf("%d %d",&a,&b);
            G[a].push_back(b);//降低复杂度,将n^2降至m
            par[b]=a;
        }

        int root;
        for(int i=1;i<=n;i++)
            if(!par[i])  {root=i;break;}

        dfs(root,1);//标记深度
        scanf("%d %d",&u,&v);

        if(dep[u]<dep[v]) swap(u,v);
        while(dep[u]>dep[v]) u=par[u];

        while(u!=v)
        {
            u=par[u];
            v=par[v];
        }
        printf("%d\n",u);
    }
    return 0;
}

  分析:最基础的LCA

POJ 1330

标签:

原文地址:http://www.cnblogs.com/smilesundream/p/5452390.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!