题意:给定有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;
}