同样的裸LCA问题!
Tarjan算法:
#include<iostream>
#include<vector>
using namespace std;
#define INF 0x7f
struct Node
{
int t,cap;
};
int dist[10010]; //到根节点的距离
int parent[10010];
bool vis[10010]; //判断是否已走过。可以用来求森林树。。。
int n,m,c;
vector<Node> map[10010];
vector<Node> qry[10010];
int ans[1000010];
int find(int x)
{
if(parent[x]!=x)
return parent[x]=find(parent[x]);
return x;
}
void dfs( int u )
{
parent[u]=u;
vis[u]=true;
for(int i=0;i<qry[u].size();i++) //到某个节点的时候,顺便求出题目所要求的查询
{
int v=qry[u][i].t;
if( parent[v] && ans[qry[u][i].cap]==-1 )
ans[qry[u][i].cap]=dist[u]+dist[v]-2*dist[find(v)];
}
for(int i=0; i<map[u].size(); i++)
{
int v=map[u][i].t;
int cap=map[u][i].cap;
if(!parent[v])
{
dist[v]+=(dist[u]+cap);
dfs( v );
parent[v]=u;
}
}
}
int main()
{
// freopen("in.txt","r",stdin);
while(scanf("%d %d %d",&n,&m,&c)==3)
{
memset(parent,0,sizeof(parent));
memset(vis,0,sizeof(vis));
memset(dist,0,sizeof(dist));
memset(ans,-1,sizeof(ans));
for(int i=0;i<10010;i++)
{
map[i].clear();
qry[i].clear();
}
for(int i=0;i<m;i++)
{
int a,b,k;
scanf("%d %d %d",&a,&b,&k);
Node x;
x.t=b; x.cap=k;
map[a].push_back(x);
x.t=a; x.cap=k;
map[b].push_back(x);
}
for(int i=0;i<c;i++)
{
int a,b;
scanf("%d %d",&a,&b);
Node x;
x.t=b; x.cap=i;
qry[a].push_back(x);
x.t=a; x.cap=i;
qry[b].push_back(x);
}
for(int i=1;i<=n;i++)
if( !vis[i] )
{
dfs( i );
memset(parent,0,sizeof(parent));
}
for(int i=0;i<c;i++)
{
if(ans[i]==-1)
printf("Not connected\n");
else printf("%d\n",ans[i]);
}
}
return 0;
}