hdu 2874 Tarjan离线 LCA

题意:给定有N个结点的森林和Q次询问,求任意给定的两个结点之间的距离。如果两个点事不连通的,输出 Not connected
用邻接表存内存会超,用vertor ;
#include<cstdio>
#include<cstring>
#include<map>
#include<vector>
#include<cmath>
#include<cstdlib>
#include<stack>
#include<queue>
#include <iomanip>
#include<iostream>
#include<algorithm>
using namespace std ;
const int N=10100;
struct node
{
	int v,w;
};
vector<node> e[10005],q[10005];
int father[N],vist[N],dis[N],LCA[1110000],flag[N];

int find(int x)
{
	return x==father[x] ? x:father[x]=find(father[x]) ;
}

void tarjan(int u,int f)
{
	vist[u]=1;
	flag[u]=f;  
	for(int i = 0 ; i<q[u].size();i++ ) 
	{
		   int v=q[u][i].v;
		   int id=q[u][i].w;
		      if(vist[v])
		      {
		        if(flag[u]==flag[v])  //两个点 是连通的 
			     LCA[id]=dis[u]+dis[v]-2*dis[find(v)];
			  }
	}
	for(int i = 0 ; i<e[u].size();i++ ) 
	{
		    int v=e[u][i].v;
		    int w=e[u][i].w;
		    if(!vist[v])
		    {
		    	   dis[v]=dis[u]+w;
		    	   tarjan(v,f);
		    	   father[v]=u;
		    }
	}
}

int main()
{
	int n,m,Q,u,v,w;
	while(~scanf("%d%d%d",&n,&m,&Q))
	{
		   for(int i=0;i<=n+1;i++)  
           father[i]=i,dis[i]=0,e[i].clear(),q[i].clear(),vist[i]=0;
           memset(LCA,-1,sizeof(LCA));
		   for(int i = 1 ; i <= m; i++)
		   {
		   	     scanf("%d%d%d",&u,&v,&w);
		   	     node tmp ;
		   	     tmp.v=v;tmp.w=w;
		   	     e[u].push_back(tmp) ;
		   	     tmp.v=u;
		   	     e[v].push_back(tmp);
		   }
		   for(int i = 1 ; i <= Q ; i++)
		   {
		   	    scanf("%d%d",&u,&v);
		   	      node tmp ;
		   	     tmp.v=v;tmp.w=i;
		   	     q[u].push_back(tmp) ;
		   	     tmp.v=u;
		   	     q[v].push_back(tmp);
		   }
	       
		   for(int i = 1 ; i <= n ; i++)
		   {
		         if(!vist[i])
		   	      tarjan(i,i);
		   }
		   
		   for(int i = 1 ; i <= Q ; i++)
		   {
		   	     if(LCA[i] == -1 ) printf("Not connected\n");
		   	     else    printf("%d\n",LCA[i]) ;
		   }		    
		
	} 
	return 0;
} 


你可能感兴趣的:(hdu 2874 Tarjan离线 LCA)