lca 解决最近公共祖先问题
怎么解决的呢, 这是个暴力dfs 记忆化,并且用利用 并查集维护;
1.dfs 是一个暴力枚举的过程,失败就返回上一层从上个状态继续枚举,没有用过的状态;
poj1470
题意
5 //五个点 从1-5 ,接下来有5行
5:(3) 1 4 2 //5节点连三个点
1:(0)
4:(0)
2:(1) 3
3:(0)
6 //六个询问
(1 5) //1 5 最近公共祖先;
(1 4)
(4 2)
(2 3)
(1 3)
(4 3)
输出
每个点 作为最近公共祖先 次数不为0的次数;
看lca 核心函数
int fin(int x)
{
int i=x;
while(f[i]!=i)
{
i=f[i];
}
return i;
}
void solve(int u)
{
int i,j;
vis[u]=1;
f[u]=u;
for(i=0; i<m; i++) // wonder have same thing
{
if(x[i]==u && vis[y[i]])
// 访问到x[i] 点且 y[i] 点被访问过,那么 y[i] 的最近祖先就是他们的祖先
cnt[fin(y[i])]++;
else if(y[i]==u
&& vis[ x[i] ] ) cnt[fin( x[i] )]++;
}
for(i=head[u]; i!=-1; i=edge[i].next) // 第一步 ,遍历所有相连的点
{
if(!vis[edge[i].v])
{
solve(edge[i].v);
f[edge[i].v]=u; 递归到最后 与该点相连的上一个点
}
}
}
完整代码
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N=1111;
int n,m,k;
int head[N*2],f[N*2],vis[N*2],x[N*N],y[N*N],
in[N*2],
cnt[N*2];
struct node
{
int v,w,next;
} edge[N*2];
void add(int u,int v,int w)
{
node E= {v,w,head[u]};
edge[k]=E;
head[u]=k++;
}
void init()
{
int i,j;
k=0;
for(i=0; i<N*2; i++)
{
head[i]=-1;
f[i]=i;
vis[i] = in[i] = cnt[i] = 0;
}
}
int fin(int x)
{
int i=x;
while(f[i]!=i)
{
i=f[i];
}
return i;
}
void solve(int u)
{
int i,j;
vis[u]=1;
f[u]=u;
for(i=0; i<m; i++) // wonder have same thing
{
if(x[i]==u && vis[y[i]])
cnt[fin(y[i])]++;
else if(y[i]==u
&& vis[ x[i] ] ) cnt[fin( x[i] )]++;
}
for(i=head[u]; i!=-1; i=edge[i].next)
{
if(!vis[edge[i].v])
{
solve(edge[i].v);
f[edge[i].v]=u;
}
}
}
int main()
{
int i,j;
while(cin>> n )
{
init();
for(i = 0 ; i < n; i++)
{
//5:(3)
int u,v,cn;
scanf("%d:(%d)",&u,&cn);
// in[u]++;
for(j = 0; j < cn; j++)
{
scanf("%d",&v);
in[v]++;
add(u ,v ,0 );
add(v ,u ,0 );
}
}
scanf("%d",&m);
for(i=0; i<m; i++)
{
char s1,s2;
scanf("%1s%d %d%1s",&s1,x+i,y+i,&s2);
}
// cout<<"llllllllllllll "<<m<<endl;
int st=0;
for(i=1; i<=n; i++)
if(!in[i])
{
st=i;
break;
}
if(st) solve(st);
// cout<<st<<endl;
// for(i=0;i<m;i++)
// printf("x= %d y= %d\n",x[i],y[i]);
for(i=1; i<=n; i++)
{
if(cnt[i])
printf("%d:%d\n",i,cnt[i]);
}
}
}