poj 1330lca模板题离线算法

#include<iostream>
#include<vector>
using namespace std;
const int MAX=10001;
int pre[MAX],visit[MAX],indegree[MAX];
vector<int>qury[MAX],tree[MAX];
int ancestor[MAX];
void init(int n) {//初始化
 int i;
 for(i=1;i<=n;i++) {
  visit[i]=0;
  pre[i]=i;
  ancestor[i]=i;
  indegree[i]=0;
  qury[i].clear();
  tree[i].clear();
 }
}
int find(int x) {
 if(x!=pre[x])
  pre[x]=find(pre[x]);//路径压缩
 return pre[x];
}
void infind(int a,int b) {//合并
   int f1=find(a);
   int f2=find(b);
   if(f1==f2)
    return ;
    pre[f1]=f2;
   return ;
}
void lca(int u) {
 int i,len;
 len=tree[u].size();
 for(i=0;i<len;i++) {
  lca(tree[u][i]);
  infind(u,tree[u][i]);//将u和其子节点联系起来
  ancestor[find(u)]=u;//将所有u的子节点的点都指向u,即u为根
 }
 visit[u]=1;
 len=qury[u].size();
 for(i=0;i<len;i++)//
  if(visit[qury[u][i]]==1) {//如果找到直接输出
   printf("%d\n",ancestor[find(qury[u][i])]);
   return ;
  }
}
int main() {
 int n,t,a,b,i;
 scanf("%d",&t);
 while(t--) {
  scanf("%d",&n);
  init(n);
  for(i=1;i<n;i++) {
   scanf("%d%d",&a,&b);
   tree[a].push_back(b);
   indegree[b]++;
  }
  scanf("%d%d",&a,&b);
  qury[a].push_back(b);
  qury[b].push_back(a);
  for(i=1;i<=n;i++)
   if(indegree[i]==0) {
    lca(i);
    break;
   }
 }
 return 0;
}
posted @ 2013-11-26 20:56  HYDhyd  阅读(152)  评论(0编辑  收藏  举报