HDU - 2586 LCA

/*

http://acm.hdu.edu.cn/showproblem.php?pid=2586
题意：

    一个村子里有n个房子，这n个房子用n-1条路连接起来，接下了有m次询问，每次询问两个房子a,b之间的距离是多少。

思路：

    很明显的最近公共祖先问题，先建一棵树，然后求出每一点i到树根的距离dis[i],然后每次询问a，b之间的距离=dis[a]+dis[b]-2*dis[LCA(a,b)];

LCA(a,b)即是a，b的最近公共祖先。。


*/
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<vector>
using namespace std;
const int maxn = 40005;
int n;

int p[maxn];                        
int inq[maxn];				
bool vis[maxn];
int dis[maxn],ans[maxn];
struct node
{
	int id,w;                                 //w在tree中为路的距离 query为提问的顺序
	node( int a,int b ){ id = a; w = b; }
};
vector<node>tree[maxn],query[maxn];       //邻接表  询问

int find( int x )                        //并查集
{
	if( p[x] == x )
		return x;
	else
		return p[x] = find(p[x]);
}
void merge( int a,int b )				 //合并并查集
{
	int x = find(a);
	int y = find(b);
	p[y] = x;
}

void LCA( int u )
{	
	vis[u] = true;
	for( int i =0; i < tree[u].size(); i ++ )
	{
		int v = tree[u][i].id;
		if( !vis[v] )
		{
			dis[v] = dis[u] + tree[u][i].w;
			LCA( v );
			merge( u,v );		//注意顺序, 先Tarjan子节点tmp, 再更新其father[tmp], 因为要保证在递推tmp所代表的子树时, father[tmp] = tmp, 而与当前子树无关. 递归回来的时候再把tmp代表的子树`并入`到当前树里
		}
	}
	
	for( int i =0; i < query[u].size(); i ++ )
	{
		int v = query[u][i].id;
		if( vis[v] )
		{
			ans[ query[u][i].w ] = dis[u] + dis[v] - 2*dis[find(v)];
		}
	}
}

void init()                             //初始化
{
	for( int i = 1; i <= n ; i ++ )
	{
		p[i] = i;
		tree[i].clear();
		query[i].clear();
	}
	memset( inq,0,sizeof(inq) );
	memset( vis,0,sizeof(vis) );
	memset( dis,0,sizeof(dis) );
}

int main()
{
	int t,m,a,b,c;
	scanf("%d",&t);
	while( t-- )
	{
		scanf("%d%d",&n,&m);
		init();
		for( int i = 1; i < n; i ++ )
		{
			scanf("%d%d%d",&a,&b,&c);
			tree[a].push_back(node(b,c));
			tree[b].push_back(node(a,c));
		}
		for( int i = 0; i < m; i ++)
		{
			scanf("%d%d",&a,&b);
			query[a].push_back( node(b,i) );
			query[b].push_back( node(a,i) );
		}
		LCA(1);
		for( int i = 0; i < m; i ++ )
		{
			printf("%d\n",ans[i]);
		}
	}
	return 0;
}

/*
/* 
    最近公共祖先lca 离线算法/Tarjan算法 
     
    方法举例说明： 
                1 
               / \ 
              2   3 
             / \ 
            4   5 
           /   / 
          7   8 
         /  
        9   
    查询(4,9)：到4时，由于vis[9]=0，所以继续；到9后，最近公共祖先就是f[4]=4（回溯的时候记录父节点）; 
    查询(9,8):深度优先搜索，所以到8时才询问；要到8必须回溯到2，在进到8这个子树，所以以记录f[9]=7;f[7]=4;f[4]=2;f[2]=2;所以find(2)=2; 
    查询(8,3):跟上条相似，必须回溯1才能到3，而find(8)=1就是最近公共祖先； 
    我们可以发现，对于查询的两点，都要在先查询到的点开始，回溯到最近公共祖先，才查询相对应的点。这也就是离线算法的思路了。。这是我自己理解的。。 
     
    下面是专业说明，反正我是没看懂。。 
     
    利用并查集优越的时空复杂度，我们可以实现LCA问题的O(n+Q)算法，这里Q表示询问的次数。 
    Tarjan算法基于深度优先搜索的框架，对于新搜索到 的一个结点，首先创建由这个结点构成的集合， 
    再对当前结点的每一个子树进行搜索，每搜索完一棵子树，则可确定子树内的LCA询问都已解决。 
    其他的LCA询问的结果必然在这个子树之外，这时把子树所形成的集合与当前结点的集合合并，并将当前结点设为这个集合的祖先。 
    之后继续搜索下一棵子树，直到当前结点的所 有子树搜索完。这时把当前结点也设为已被检查过的， 
    同时可以处理有关当前结点的LCA询问，如果有一个从当前结点到结点v的询问， 
    且v已被检查过，则由于 进行的是深度优先搜索，当前结点与v的最近公共祖先一定还没有被检查， 
    而这个最近公共祖先的包涵v的子树一定已经搜索过了，那么这个最近公共祖先一定是v 所在集合的祖先。 
*/  
*/