POJ 1330 Nearest Common Ancestors(tarjan LCA 算法)
關(guān)于LCA和RMQ問(wèn)題
|
一、最近公共祖先(Least Common Ancestors) 對(duì)于有根樹(shù)T的兩個(gè)結(jié)點(diǎn)u、v,最近公共祖先LCA(T,u,v)表示一個(gè)結(jié)點(diǎn)x,滿足x是u、v的祖先且x的深度盡可能大。另一種理解方式是把T理解為一個(gè)無(wú)向無(wú)環(huán)圖,而LCA(T,u,v)即u到v的最短路上深度最小的點(diǎn)。 這里給出一個(gè)LCA的例子: 例一 對(duì)于T=<V,E> 則有: LCA(T,5,2)=1 RMQ問(wèn)題是指:對(duì)于長(zhǎng)度為n的數(shù)列A,回答若干詢問(wèn)RMQ(A,i,j)(i,j<=n),返回?cái)?shù)列A中下標(biāo)在[i,j]里的最小值下標(biāo)。這時(shí)一個(gè)RMQ問(wèn)題的例子: 例二 對(duì)數(shù)列:5,8,1,3,6,4,9,5,7 有: RMQ(2,4)=3 RMQ問(wèn)題與LCA問(wèn)題的關(guān)系緊密,可以相互轉(zhuǎn)換,相應(yīng)的求解算法也有異曲同工之妙。 下面給出LCA問(wèn)題向RMQ問(wèn)題的轉(zhuǎn)化方法。 對(duì)樹(shù)進(jìn)行深度優(yōu)先遍歷,每當(dāng)“進(jìn)入”或回溯到某個(gè)結(jié)點(diǎn)時(shí),將這個(gè)結(jié)點(diǎn)的深度存入數(shù)組E最后一位。同時(shí)記錄結(jié)點(diǎn)i在數(shù)組中第一次出現(xiàn)的位置(事實(shí)上就是進(jìn)入結(jié)點(diǎn)i時(shí)記錄的位置),記做R[i]。如果結(jié)點(diǎn)E[i]的深度記做D[i],易見(jiàn),這時(shí)求LCA(T,u,v),就等價(jià)于求E[RMQ(D,R[u],R[v])],(R[u]<R[v])。例如,對(duì)于第一節(jié)的例一,求解步驟如下: 數(shù)列E[i]為:1,2,1,3,4,3,5,3,1 于是有: LCA(T,5,2) = E[RMQ(D,R[2],R[5])] = E[RMQ(D,2,7)] = E[3] = 1 易知,轉(zhuǎn)化后得到的數(shù)列長(zhǎng)度為樹(shù)的結(jié)點(diǎn)數(shù)的兩倍加一,所以轉(zhuǎn)化后的RMQ問(wèn)題與LCA問(wèn)題的規(guī)模同次 |
數(shù)據(jù)很水,但是算法絕對(duì)經(jīng)典,題目要求兩個(gè)節(jié)點(diǎn)的最近公共祖先,這個(gè)tarjan算法使用了并查集+dfs的操作。這個(gè)算法研究了我兩個(gè)小時(shí),開(kāi)始的時(shí)候在網(wǎng)上查找資料,寫(xiě)的都不盡人意,看不明白,后來(lái)干脆直接看代碼,模擬了一遍,搞懂了。。。看來(lái)計(jì)算機(jī)的世界,只有代碼才是王道!中間的那個(gè)并查集操作的作用,只是將已經(jīng)查找過(guò)的節(jié)點(diǎn)捆成一個(gè)集合然后再指向一個(gè)公共的祖先。另外,如果要查詢LCA(a,b),必須把(a,b)和(b,a)都加入鄰接表。
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;
#define MAX 10001
int f[MAX];
int r[MAX];
int indegree[MAX];
int visit[MAX];
vector<int> hash[MAX],Qes[MAX];
int ancestor[MAX];

void init(int n)

{
int i;
for(i=1;i<=n;i++)
{
r[i]=1;
f[i]=i;
indegree[i]=0;
visit[i]=0;
ancestor[i]=0;
hash[i].clear();
Qes[i].clear();
}
}

int find(int n)

{
if(f[n]==n)
return n;
else
f[n]=find(f[n]);
return f[n];
}//查找函數(shù),并壓縮路徑

int Union(int x,int y)

{
int a=find(x);
int b=find(y);
if(a==b)
return 0;
else if(r[a]<=r[b])
{
f[a]=b;
r[b]+=r[a];
}
else
{
f[b]=a;
r[a]+=r[b];
}
return 1;
}//合并函數(shù),如果屬于同一分支則返回0,成功合并返回1

void LCA(int u)

{
ancestor[u]=u;
int i,size = hash[u].size();
for(i=0;i<size;i++)
{
LCA(hash[u][i]);
Union(u,hash[u][i]);
ancestor[find(u)]=u;
}
visit[u]=1;
size = Qes[u].size();
for(i=0;i<size;i++)
{
if(visit[Qes[u][i]]==1)
{
printf("%d\n",ancestor[find(Qes[u][i])]);
return;
}
}
}

int main()

{
int testcase;
int n;
int i,j;
scanf("%d",&testcase);
for(i=1;i<=testcase;i++)
{
scanf("%d",&n);
init(n);
int s,t;
for(j=1;j<=n-1;j++)
{
scanf("%d%d",&s,&t);
hash[s].push_back(t);
indegree[t]++;
}
scanf("%d%d",&s,&t);
Qes[s].push_back(t);
Qes[t].push_back(s);
for(j=1;j<=n;j++)
{
if(indegree[j]==0)
{
LCA(j);
break;
}
}
}
return 0;
}


還有就是,如果這個(gè)樹(shù)的根節(jié)點(diǎn)不確定,邊沒(méi)有方向可言的話,又應(yīng)該如何做呢?(找不到根了。。。)
如果要找兩個(gè)節(jié)點(diǎn)之間的一條通路 又該怎么辦?
看來(lái),這個(gè)問(wèn)題還有繼續(xù)研究下去的必要。。。
———————————————————————傳說(shuō)中的分割線——————————————————————————
指針版:
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;
#define MAX 10001
struct node

{
int a;
node *next;
}hash[MAX];
node Qes[MAX];
void Add(node hash[],int a,int b)

{
node *p=&hash[a];
node *q=p->next;
node *r=new node;
r->a=b;
p->next=r;
r->next=q;
}
int f[MAX];
int r[MAX];
int indegree[MAX];
int visit[MAX];
int ancestor[MAX];

void init(int n)

{
int i;
for(i=1;i<=n;i++)
{
r[i]=1;
f[i]=i;
indegree[i]=0;
visit[i]=0;
ancestor[i]=0;
hash[i].next=NULL;
Qes[i].next=NULL;
}
}

int find(int n)

{
if(f[n]==n)
return n;
else
f[n]=find(f[n]);
return f[n];
}//查找函數(shù),并壓縮路徑

int Union(int x,int y)

{
int a=find(x);
int b=find(y);
if(a==b)
return 0;
else if(r[a]<=r[b])
{
f[a]=b;
r[b]+=r[a];
}
else
{
f[b]=a;
r[a]+=r[b];
}
return 1;
}//合并函數(shù),如果屬于同一分支則返回0,成功合并返回1

void LCA(int u)

{
ancestor[u]=u;
node *p;
for(p=hash[u].next;p!=NULL;p=p->next)
{
LCA(p->a);
Union(u,p->a);
ancestor[find(u)]=u;
}
visit[u]=1;
for(p=Qes[u].next;p!=NULL;p=p->next)
{
if(visit[p->a]==1)
{
printf("%d\n",ancestor[find(p->a)]);
return;
}
}
}

int main()

{
int testcase;
int n;
int i,j;
scanf("%d",&testcase);
for(i=1;i<=testcase;i++)
{
scanf("%d",&n);
init(n);
int s,t;
for(j=1;j<=n-1;j++)
{
scanf("%d%d",&s,&t);
Add(hash,s,t);
indegree[t]++;
}
scanf("%d%d",&s,&t);
Add(Qes,s,t);
Add(Qes,t,s);
for(j=1;j<=n;j++)
{
if(indegree[j]==0)
{
LCA(j);
break;
}
}
}
return 0;
}
汗,寫(xiě)了個(gè)指針版運(yùn)行速度居然比vector還慢,太打擊我對(duì)指針的信心了。。。可能是數(shù)據(jù)太弱了吧,指針的優(yōu)勢(shì)沒(méi)有發(fā)揮出來(lái)。。。
——————————————————傳說(shuō)中的分割線——————————————————————————————————
終于用RMQ過(guò)了,起初一直RE,原來(lái)是RM Q里面的數(shù)組開(kāi)小了。。。不過(guò)感覺(jué)時(shí)間效率不是很高,110MS,應(yīng)該是詢問(wèn)的次數(shù)太少的原因吧。
RMQ應(yīng)付的應(yīng)該是詢問(wèn)次數(shù)非常巨大的情況。而且它是在線的算法,可以按順序輸出結(jié)果,這樣才能AC題目啊。
#include<iostream>
#include<math.h>
#include<vector>
using namespace std;
vector<int>hash[10001];//用vector代替鄰接表
#define MAXN 10001
#define mmin(seq, a, b) ((seq[a] < seq[b]) ? (a) : (b))
int indegree[MAXN];
/**//**//**///// DP status
int fij[100000][100];
template <typename T>
void st(T seq[], int n)//預(yù)處理

{
memset(fij, 0, 100 * MAXN * sizeof(int));
int k = (int)(log((double)n) / log(2.0)); 
/**//**//**/////初始狀態(tài)
for(int i = 0; i < n; i++)
fij[i][0] = i; 
/**//**//**/////遞推計(jì)算狀態(tài)
for(int j = 1; j <= k; j++)
{
for(int i = 0; i + (1 << j) - 1 < n; i++)
{
//
int m = i + (1 << (j - 1));
//fij[i][j] = seq[fij[i][j - 1]] < seq[fij[m][j - 1]] ? fij[i][j - 1] : fij[m][j - 1];
fij[i][j] = mmin(seq, fij[i][j - 1], fij[m][j - 1]);
}
}
}
template <typename T>
int RMQ(T seq[], int i, int j)//求解RMQ

{
/**//**//**///// 用對(duì)2去對(duì)數(shù)的方法求出k
int k = (int)(log(double(j - i + 1)) / log(2.0)); 
/**//**//**/////
//int t = seq[fij[i][k]] < seq[fij[j - (1 << k) + 1][k]] ? fij[i][k] : fij[j - (1 << k) + 1][k];
int t = mmin(seq, fij[i][k], fij[j - (1 << k) + 1][k]);
return t;
}
int E[MAXN*2+1];
int R[MAXN*2+1];
int D[MAXN*2+1];
int p=0;
void dfs(int r,int deep)//深搜,算出E,R,D數(shù)組

{
p++;
E[p]=r;
D[p]=deep;
R[r]=p;
int i;
int size=hash[r].size();
for(i=0;i<size;i++)
{
dfs(hash[r][i],deep+1);
p++;
E[p]=r;
D[p]=deep;
}
}
int main()

{
int testcase;
int n;
int i,j;
int s,t;
scanf("%d",&testcase);
for(i=1;i<=testcase;i++)
{
scanf("%d",&n);
for(j=1;j<=n;j++)
{
indegree[j]=0;
hash[j].clear();
}
p=0;
for(j=1;j<n;j++)
{
int a,b;
scanf("%d%d",&a,&b);
hash[a].push_back(b);
indegree[b]++;
}
scanf("%d%d",&s,&t);
int root;
for(j=1;j<=n;j++)
{
if(indegree[j]==0)
{
root=j;
break;
}
}
dfs(root,0);
st(D,2*n+2);
if(R[s]<R[t])
printf("%d\n",E[ RMQ(D,R[s],R[t]) ] );
else
printf("%d\n",E[ RMQ(D,R[t],R[s]) ] );
}
return 0;
}

posted on 2009-09-21 22:24 abilitytao 閱讀(3567) 評(píng)論(1) 編輯 收藏 引用

