Nearest Common Ancestors POJ

Nearest Common Ancestors

POJ - 1330

题意：找两个点公共祖先，裸题。

#include <cstdio>
#include <cstring>
const int maxn=10010;

int f[maxn],vis[maxn];

int main(){
    int n,t;
   // freopen("in.txt","r",stdin);
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        int u,v;
        memset(vis,0,sizeof(vis));
        memset(f,0,sizeof(f));
        for(int i=1;i<n;i++){
            scanf("%d%d",&u,&v);
            f[v]=u;
        }
        scanf("%d%d",&u,&v);
        int x=u;
        while(x){
            vis[x]=1;
            x=f[x];
        }
        x=v;
        while(x&&!vis[x]){
            x=f[x];
        }
        printf("%d
",x);
    }
}

倍增法

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int maxn=10010;

int tot;
int fr[maxn],dep[maxn<<1],p[maxn<<1];
int dp[maxn<<1][30],deg[maxn];
struct Edge{
    int v,nex;
}e[maxn<<1];
int head[maxn];
int cnt=0;
void init(){
    tot=0;
    memset(head,-1,sizeof(head));
    memset(deg,0,sizeof(deg));
    cnt=0;
}
void add(int u,int v){
    e[cnt].v=v;
    e[cnt].nex=head[u];
    head[u]=cnt++;
}

void dfs(int u,int f,int d){
    tot++;
    p[tot]=u;
    dep[tot]=d;
    fr[u]=tot;
    for(int i=head[u];~i;i=e[i].nex){
        int v=e[i].v;
        if(v==f) continue;
        dfs(v,u,d+1);
        tot++;
        p[tot]=u;
        dep[tot]=d;
    }
    return ;
}

void RMQ_init(int n){
    int k=1;
    while((1<<k+1)<n) k++;
    for(int i=1;i<=n;i++) dp[i][0]=i;
    for(int j=1;j<=k;j++){
        for(int i=1;i+(1<<j)-1<=n;i++){
            int a=dp[i][j-1];
            int b=dp[i+(1<<j-1)][j-1];
            if(dep[a]<dep[b]) dp[i][j]=a;
            else dp[i][j]=b;
        }
    }
    return ;
}

int rmq(int u,int v){
    int k=1;
    while((1<<k+1)<v-u+1) k++;
    int a=dp[u][k];
    int b=dp[v-(1<<k)+1][k];
    if(dep[a]<dep[b]) return a;
    else return b;
}

int lca(int u,int v){
    u=fr[u];
    v=fr[v];
    if(u>v) swap(u,v);
    int rt=rmq(u,v);
    return p[rt];
}

int main(){
    int n,t;
   // freopen("in.txt","r",stdin);
    scanf("%d",&t);
    while(t--){
        init();
        scanf("%d",&n);
        int u,v;
        for(int i=1;i<n;i++){
            scanf("%d%d",&u,&v);
            add(u,v);
            deg[v]++;
        }
        for(int i=1;i<=n;i++) if(!deg[i]) dfs(i,-1,1);
        RMQ_init(tot);
        scanf("%d%d",&u,&v);
        printf("%d
",lca(u,v));
    }
    return 0;
}