最近公共祖先 LCA

Tarjan算法

前向星型

#include
#define N 500000

struct Edge{
	int to,next,w;
}edge[N],q[N];  //q数组存储查询的两点


int num_edge,num_q,head[N],qhead[N];
int pre[N],vis[N];//pre[i]存储i的父节点
 
//建立链表 添加边 
void add_edge(int from,int to)
{
	edge[++num_edge].next=head[from];//以from结点出发的上一条边在数组中的编号 
	edge[num_edge].to=to;
	head[from]=num_edge;
}

void add_q(int from,int to)
{
	q[++num_q].next=qhead[from];
	q[num_q].to=to;
	qhead[from]=num_q;
}

int find(int x)//寻找x的父亲结点 
{
	while(pre[x]!=x)
	  x=pre[x];
	return x;
}

int dfs(int x)
{
	pre[x]=x;
	vis[x]=1;
	//遍历与x相连的结点 
	for(int k=head[x];k;k=edge[k].next){
		int v=edge[k].to;
		if(!vis[v]){//如果未被搜索 
			dfs(v);//以该结点为根节点,继续搜索
			pre[v]=x;//把v的父亲结点设为x 
		}
	}
	//搜索包含结点x的所有查询 
	for(int k=qhead[x];k;k=q[k].next){
		int v=q[k].to;
		if(vis[v]){//如果另一结点已被搜索过 
			q[k].w=find(v);//把另一节点的祖先设为这两个节点的最近公共祖先 
			if(k%2)//由于将每一组查询变为两组,所以2n-1和2n的结果是一样的
			  q[k+1].w=q[k].w;
			else
			  q[k-1].w=q[k].w; 
		}
	}
}
int main()
{
	int n,m,p,x,y,i;
	scanf("%d%d%d",&n,&m,&p);//输入结点数,查询数和根结点
	for(i=1;i

暴力求LCA

//暴力求LCA
#include
#include
#include

using namespace std;
const int N=10010;
vectorg[N];//邻接表
int depth[N],f[N],in[N];//depth存储各结点的深度,f存储各结点的父节点 

void dfs(int rt,int fa)//分别表示本结点和本结点的父亲结点 
{
	depth[rt]=depth[fa]+1;
	f[rt]=fa;
	for(int i=0;idepth[b])
	  swap(a,b);
	while(depth[b]>depth[a])
	  b=f[b];
	while(a!=b)
	  a=f[a],b=f[b];
	return a;
}


int main()
{
	int t,n,a,b;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=0;i

倍增算法求LCA

//倍增算法求LCA
#include
#include
#include

using namespace std;
const int N=10010;
vectorg[N];//邻接表
int depth[N],f[N][20],in[N];//depth存储各结点的深度,f存储各结点的父节点 

void dfs(int rt,int fa)//分别表示本结点和本结点的父亲结点 
{
	depth[rt]=depth[fa]+1;
	f[rt][0]=fa;
	for(int i=1;i<20;i++)
	  f[rt][i]=f[f[rt][i-1]][i-1];
	for(int i=0;idepth[b])
	  swap(a,b);
	for(int i=19;i>=0;i--)
	  if(depth[b]-(1<=depth[a])
	    b=f[b][i];
	if(a==b)
	  return a;
	for(int i=19;i>=0;i--)
	  if(f[b][i]!=f[a][i])
	    a=f[a][i],b=f[b][i];
	return f[b][0];
}

int main()
{
	int t,n,a,b;
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=0;i

题目集锦

 POJ1330 Nearest Common Ancestors【LCA】

你可能感兴趣的:(数据结构)