Connections between cities
题意:求连接两城市的最短距离
题目思路:LCA Tarjan
解题方法:通过Tarjan求距离
需注意的问题:存在森林,所以对于每棵树都得进行计算
存在未连通,只需先判断节点的根节点(并查集find)是否相同即可
内存
29432K险过,需用邻接表模拟,vector会爆
AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define N 10010
#define NN 2000010
struct Edge
{
int v,len,next;
} edge[N*2];
int tot,head[N];
void add(int a,int b,int c)
{
edge[tot].v=b;
edge[tot].len=c;
edge[tot].next=head[a];
head[a]=tot++;
edge[tot].v=a;
edge[tot].len=c;
edge[tot].next=head[b];
head[b]=tot++;
}
struct Query{
int v,index,next;
};
Query Q[NN];
int q_tot,q_head[N];
void add_ques(int a,int b,int index)
{
Q[q_tot].v=b;
Q[q_tot].index=index;
Q[q_tot].next=q_head[a];
q_head[a]=q_tot++;
Q[q_tot].v=a;
Q[q_tot].index=index;
Q[q_tot].next=q_head[b];
q_head[b]=q_tot++;
}
bool vis[N],root[N];
int pre[N],dis[N],ans[NN/2];
int n,q;
void init(){
for(int i=0;i<=n;i++){
pre[i]=i;
vis[i]=false;
}
tot=q_tot=0;
memset(q_head,-1,sizeof(q_head));
memset(head,-1,sizeof(head));
memset(ans,-1,sizeof(ans));
memset(root,true,sizeof(root));
}
int find(int x){
return pre[x]==x?pre[x]:pre[x]=find(pre[x]);
}
void Union(int x,int y){
x=find(x);
y=find(y);
if(x!=y){
pre[y]=x;
}
}
void Tarjan(int cur,int val){
dis[cur]=val;
vis[cur]=true;
for(int j=head[cur]; j!=-1; j=edge[j].next)
{
int v=edge[j].v;
if(vis[v]) continue;
Tarjan(v,val+edge[j].len);
Union(cur,v);
}
for(int j=q_head[cur]; j!=-1; j=Q[j].next)
{
int v=Q[j].v;
if(!vis[v]) continue;
ans[Q[j].index]=dis[v]+dis[cur]-2*dis[find(v)];
}
}
int main(){
int T,m;
// scanf("%d",&T);
while(scanf("%d%d%d",&n,&m,&q)!=EOF){
int a,b,c;
init();
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
root[b]=false;
}
for(int i=0;i<q;i++){
scanf("%d%d",&a,&b);
add_ques(a,b,i);
}
///对于森林的每颗树进行计算
for(int i=1;i<=n;i++){
if(!vis[i]&&root[i]){
Tarjan(i,0);
}
}
for(int i=0;i<q;i++){
///查找两节点的根节点,如果不相同则未连同
if(find(Q[i*2].v)!=find(Q[2*i+1].v)) printf("Not connected\n");
else printf("%d\n",ans[i]);
}
}
return 0;
}