POJ 1330 Nearest Common Ancestors【tarjan】【简单LCA】【模板题】【并查集】

题目链接:http://poj.org/problem?id=1330
题目大意:给出n个父子关系,不会出现回路,求最原始祖先(根节点)。
照大佬的板子自己手码了一遍,还挺长,,,
AC代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<limits>
#include<vector>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
const int maxx=10010;
int F[maxx];
int r[maxx];
bool vis[maxx];
int anc[maxx];
struct node
{
    int to,nxt;
}edge[2*maxx];
int head[maxx];
int tol;
void addedge(int a,int b)
{
     edge[tol].to=b;
     edge[tol].nxt=head[a];
     head[a]=tol++;
     edge[tol].to=a;
     edge[tol].nxt=head[b];
     head[b]=tol++;
}
struct Query
{
    int q;
    int nxt;
    int index;
}query[2*maxx];
int ans[2*maxx];
int cnt;
int h[maxx];
int tt;
int Q;
void add_query(int a,int b,int i)
{
     query[tt].q=b;
     query[tt].nxt=h[a];
     query[tt].index=i;
     h[a]=tt++;
     query[tt].q=a;
     query[tt].nxt=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]=0;
          anc[i]=0;
          tol=0;
          tt=0;
          cnt=0;
     }
     memset(head,-1,sizeof(head));
     memset(h,-1,sizeof(h));
}
int findd(int x)
{
     if(F[x]==-1)
          return x;
     return F[x]=findd(F[x]);
}
void Union(int x,int y)
{
     int t1=findd(x);
     int t2=findd(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)
{
     anc[u]=u;
     vis[u]=1;
     for(int i=head[u];i!=-1;i=edge[i].nxt)
     {
          int v=edge[i].to;
          if(vis[v])
                    continue;
          LCA(v);
          Union(u,v);
          anc[findd(u)]=u;
     }
     for(int i=h[u];i!=-1;i=query[i].nxt)
     {
          int v=query[i].q;
          if(vis[v])
          {
              ans[query[i].index]=anc[findd(v)];
              cnt++;
          }
     }
}
bool flag[maxx];
int main()
{
    int t;
    int u,v;
    int N;
    scanf("%d",&t);
    while(t--)
    {
         scanf("%d",&N);
         init(N);
         memset(flag,0,sizeof(flag));
         for(int i=1;i<N;i++)
         {
              scanf("%d%d",&u,&v);
              flag[v]=1;
              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",ans[i]);
         }
         printf("\n");
    }
    return 0;
}

我一定可以的!!!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值