[省选前题目整理][POJ 1330]Nearest Common Ancestors(倍增法求LCA)

题目链接

http://poj.org/problem?id=1330

思路

非常裸的倍增LCA练手题。。。

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 12000

using namespace std;

struct edge
{
    int u,v,next;
}edges[MAXN];

int head[MAXN],nCount=0;

void AddEdge(int U,int V)
{
    edges[++nCount].u=U;
    edges[nCount].v=V;
    edges[nCount].next=head[U];
    head[U]=nCount;
}

int fa[MAXN][20],n,depth[MAXN];

void DFS(int u)
{
    for(int p=head[u];p!=-1;p=edges[p].next)
    {
        int v=edges[p].v;
        depth[v]=depth[u]+1;
        fa[v][0]=u;
        DFS(v);
    }
}

void prework()
{
    for(int i=1;i<=18;i++)
    {
        for(int j=1;j<=n;j++)
            fa[j][i]=fa[fa[j][i-1]][i-1];
    }
}

int LCA(int a,int b)
{
    if(depth[a]<depth[b]) swap(a,b); //保证a比b深度大
    for(int i=18;i>=0;i--)
    {
        //if(!fa[a][i]) continue;
        if(depth[fa[a][i]]<=depth[b]) continue;
        a=fa[a][i];
    }
    if(depth[a]!=depth[b]) a=fa[a][0];
    if(a==b) return a;
    for(int i=18;i>=0;i--)
    {
        if(!fa[a][i]||!fa[b][i]) continue;
        if(fa[a][i]==fa[b][i]) continue;
        a=fa[a][i],b=fa[b][i];
    }
    a=fa[a][0],b=fa[b][0];
    return a;
}

int inDegree[MAXN];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(inDegree,0,sizeof(inDegree));
        nCount=0;
        memset(head,-1,sizeof(head));
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            AddEdge(u,v);
            inDegree[v]++;
        }
        for(int i=1;i<=n;i++)
            if(!inDegree[i])
                DFS(i);
        prework();
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",LCA(a,b));
    }
    return 0;
}