poj1330 LCA离线tarjan算法 复杂度O(n+q)

http://poj.org/problem?id=1330

题意：给出一个森林，求两节点的的lca

解题思路：

lca的离线tarjan算法，在dfs时不断将子节点并到父结点的集合中，在搜索另一棵树前，要重置f数组

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<bitset>
#include<utility.h>
using namespace std;
#define N 10005
vector<int>tree[N];//图的邻接表
vector<pair<int,int> >Q[N];//询问
int ans[N];//保存每个询问的结果
int indegree[N];//保存每个节点的入度
int f[N];
int find(int x)//找父根节点
{
	if(f[x]!=x)
	{f[x]=find(f[x]);}
	return f[x];
}
void mark(int a,int b)
{
    f[find(b)] = find(a);
}
void LCA(int u) 
{
    f[u]=u;
	for(int i=0; i<tree[u].size(); i++) //未visit子节点
    {
        LCA(tree[u][i]);
        mark(u, tree[u][i]);//将子节点并到父结点的集合中
    }
	for(int i=0; i<Q[u].size(); i++)//求当前节点与有关的节点的最近公共祖先
	{
		if(f[Q[u][i].first]!=-1)//如果另一个节点也已处理过
		{
		    ans[Q[u][i].second]=find(Q[u][i].first);
		}
	}
}
void callLCA(int n)
{
    for(int i=1; i<=n; i++)
	{
		if(indegree[i]==0)//寻找根节点，根节点的入度为0
		{
		    LCA(i);
		    memset(f,-1,sizeof(f));
		}
	}
}
void init(int n)
{
    memset(f,-1,sizeof(f));
    memset(indegree,0,sizeof(indegree));
    memset(ans,-1,sizeof(ans));
    for(int i=0;i<=n;i++)
    {
        tree[i].clear();
        Q[i].clear();
    }
}
int main()
{
	int i,j,k;
	int t,n,m;
	scanf("%d",&t);
	while(t--)
	{
	    scanf("%d",&n);
	    init(n);
        int from,to;
        m=n-1;//边数
	    for(i=1;i<=m;i++)
	    {
	        scanf("%d%d",&from,&to);//有from->to的一条边
	        tree[from].push_back(to);
	        indegree[to]++;
	    }
	    int q=1;//问题数
	    pair<int,int>p;
	    for(i=1;i<=q;i++)
	    {
	        scanf("%d%d",&from,&to);
	        p=make_pair(to,i);
            Q[from].push_back(p);
            p=make_pair(from,i);
            Q[to].push_back(p);
	    }
	    callLCA(n);
        for(i=1;i<=q;i++)
        {
            if(ans[i]==-1)
            {printf("not connect");}
            else
            {printf("%d\n",ans[i]);}
        }
	}
}
/*
input:
1
5
2 3
3 4
3 1
1 5
3 5
output:
3
*/

参考资料：

http://dongxicheng.org/structure/lca-rmq/

http://blog.csdn.net/smallacmer/article/details/7432664

http://blog.csdn.net/cxllyg/article/details/7635992