poj1330 LCA 屯個模板

//給一棵樹,最後問兩個點的lca是誰。
#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;

const int maxn=10005;
struct node{
    int v,next;
};
node edge[maxn];
int head[maxn],cnt;
void add(int a,int b){
    edge[cnt].v=b;
    edge[cnt].next=head[a];
    head[a]=cnt++;
}


int deep[maxn<<1];//第i個訪問的點的深度
int index[maxn<<1];//第i個訪問的節點的編號
int first[maxn];//第i個節點最早出現的時候
int dp[maxn<<1][16];
int in[maxn];
int time;
void dfs(int u,int dep){
    index[++time]=u;
    deep[time]=dep;
    if(first[u]==0)
        first[u]=time;
    for(int i=head[u];i!=-1;i=edge[i].next){
            int v=edge[i].v;
            dfs(v,dep+1);
            index[++time]=u;
            deep[time]=dep;
        }
}
void rmq_init(int n){
    int i,j;
    for(int i=1;i<=n;i++)
        dp[i][0]=i;
    for(int j=1;1<<j<=n;j++){
        int k=1<<(j-1);
        for(int i=1;i+k<n;i++){
            if(deep[dp[i][j-1]]<=deep[dp[i+k][j-1]])
                dp[i][j]=dp[i][j-1];
            else
                dp[i][j]=dp[i+k][j-1];
        }
    }
}
int rmq(int a,int b){
    int dis=abs(b-a)+1;
    int k=log(double(dis))/log(2.0);
    if(deep[dp[a][k]]<=deep[dp[b-(1<<k)+1][k]]){
        return dp[a][k];
    }
    else
        return dp[b-(1<<k)+1][k];
}

int LCA(int u,int v){
    return first[u]<=first[v]?index[rmq(first[u],first[v])] :index[rmq(first[v],first[u])];
}

void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
    memset(first,0,sizeof(first));
    memset(in,0,sizeof(in));
}

int main(){
    int n,m,q;
    int root=0;
    int u,v,d;
    int T;
    scanf("%d",&T);
    while(T--){
            scanf("%d",&n);
            int tmp=n;
            init();
            tmp--;
            while(tmp--){
                scanf("%d%d",&u,&v);
                add(u,v);
                in[v]++;
            }
            time=0;
            for(int i=1;i<=n;i++){
                if(in[i]==0){
                    dfs(i,0);
                    break;
                }
            }
            rmq_init(time);
            scanf("%d%d",&u,&v);
           printf("%d\n",LCA(u,v));
    }
return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章