【最近公共祖先(LCA)】


/*

题目大意:开始让你构造一棵树,然后有Q对u,v,最后让你求出每对u,v的最近祖先的编号和次数。
POJ  1470
*/

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<cmath>
#define N 4010
#define M 4010
using namespace std;
int visit[M], level[M], first[N]; //深搜步骤(祖先标号)、层、首次出现
int dp[20][N]; //RMQ-dp处理
vector<int>son[N]; //记录孩子
int in[N]; //入度
int top;
int res[N]; //结果
void init(int n){
    top=1;
    memset(in,0,sizeof(in));
    memset(first,-1,sizeof(first));
    memset(res,0,sizeof(res));
    for(int i=1;i<=n;i++)
        son[i].clear();
}
void DFS(int ceng,int now){
    if(first[now]==-1)
        first[now]=top;
    level[top]=ceng;
    visit[top++]=now;
    if(son[now].size()==0)
        return ;
    for(int i=0;i<son[now].size();i++){
        DFS(ceng+1,son[now][i]);
        level[top]=ceng;
        visit[top++]=now;
    }
}
void RMQ(int n){//预处理
    for(int i=1;i<=n;i++)
        dp[0][i]=level[i];
    for(int i = 1; (1 << i) <= n; ++i)
        for(int j = 1; j <= n + 1 - (1 << i); ++j)
            dp[i][j] = min(dp[i - 1][j], dp[i - 1][j + (1 << i >> 1)]);
}
int ac(int start,int endx){  //返回最近公共祖先
    int k = (int)(log((double)(endx - start + 1.0)) / log(2.0));
    int temp = min(dp[ k ][ start ], dp[ k ][ endx - (1 << k) + 1 ]);
    for(int i = start; i <= endx; ++i) //找下标
        if(level[i] == temp)
            return visit[i];
}

int main(){
    int vnum;
    int a,num,b;
    int root;
    int query,start,endx;
    while(~scanf("%d",&vnum)){
        init(vnum);
        for(int i=1;i<=vnum;i++){
            scanf("%d:(%d)",&a,&num);
            while(num--){
                scanf("%d",&b);
                son[a].push_back(b);
                in[b]++;
            }
        }
        for(root=1;in[root];++root);//找根
        DFS(0,root);//深搜
        RMQ(2*vnum-1);//O(nlogn)预处理
        scanf("%d",&query);
        while(query--){
            scanf(" (%d%d) ",&start,&endx);
            start=first[start],endx=first[endx];//找到在level中的位置
            if(start>endx)//保证前小后大
                swap(start,endx);
            res[ac(start,endx)]++;
        }
        for(int i=1;i<=vnum;i++)//求出每个数作为祖先的次数
            if(res[i])
                printf("%d:%d\n",i,res[i]);
    }
}



/*
POJ 1330
离线算法
给定一棵树,求两个结点的最近公共祖先。
*/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN =10010;
int F[MAXN];//并查集
int r[MAXN];//并查集中集合的个数
bool vis[MAXN];//访问标记
int ancestor[MAXN];//祖先
struct Node{
    int to,next;
}edge[MAXN*2];
int head[MAXN];
int tol;
void addedge(int a,int b){
    edge[tol].to=b;
    edge[tol].next=head[a];
    head[a]=tol++;
    edge[tol].to=a;
    edge[tol].next=head[b];
    head[b]=tol++;
}
struct Query{
    int q,next;
    int index;//查询编号
}query[MAXN*2];//查询结果
int answer[MAXN*2];
int cnt;
int h[MAXN];
int tt;
int Q;//查询个数
void add_query(int a,int b,int i){
    query[tt].q=b;
    query[tt].next=h[a];
    query[tt].index=i;
    h[a]=tt++;
    query[tt].q=a;
    query[tt].next=h[b];
    query[tt].index=i;
    h[b]=tt++;
}
void init(int n){
    for(int i=1;i<=n;i++){
        F[i]=-1;
        r[i]=1;
        vis[i]=false;
        ancestor[i]=0;
        tol=0;
        cnt=0;
    }
    memset(head,-1,sizeof(head));
    memset(h,-1,sizeof(h));
}
int find_set(int x){
    if(F[x]==-1)
        return x;
    return F[x]=find_set(F[x]);
}
void union_set(int x,int y){
    int t1=find_set(x);
    int t2=find_set(y);
    if(t1!=t2){
        if(r[t1]<=r[t2]){
            F[t1]=t2;
            r[t2]+=r[t1];
        }
        else{
            F[t2]=t1;
            r[t1]+=r[t2];
        }
    }
}
void LCA(int u){
    ancestor[u]=u;
    vis[u]=true;//这个一定要放在前面
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(vis[v]) continue;
        LCA(v);
        union_set(u,v);
        ancestor[find_set(u)]=u;
    }
    for(int i=h[u];i!=-1;i=query[i].next){
        int v=query[i].q;
        if(vis[v]){
            answer[query[i].index]=ancestor[find_set(v)];
            cnt++;//已经找到的答案数
        }
    }
}
bool flag[MAXN];
int main(){
    int T;
    int N;
    int u,v;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&N);
        init(N);
        memset(flag,false,sizeof(flag));
        for(int i=1;i<N;i++){
            scanf("%d%d",&u,&v);
            flag[v]=true;
            addedge(u,v);
        }
        Q=1;
        scanf("%d%d",&u,&v);
        add_query(u,v,0);
        int root;
        for(int i=1;i<=N;i++){
            if(!flag[i]){
                root=i;
                break;
            }

        }
        LCA(root);
        for(int i=0;i<Q;i++)
            printf("%d\n",answer[i]);
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值