Sample Input
7 6
1 6 13 E
6 3 9 E
3 5 7 S
4 1 3 N
2 4 20 W
4 7 2 S
3
1 6
1 4
2 6
Sample Output
13
3
36
题目大意:求两个节点之间的距离。
解题思路:可以用dfs遍历出每个节点到根节点的距离,然后找到两个节点的公共祖先,两个节点到根节点的距离之和减去最近公共祖先节点到根节点距离的两倍,就是两个根节点的距离之和了。dis=dis[u]+dis[v]-2*dis[find(v)]。中间会用到并查集,很容易理解的,百度一下你就知道。
Code:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 1e5+7;
int n,m,cnt,qcnt,head[N],qhead[N],fa[N],vis[N],dis[N];
struct node{
int w,to,next;
int dis,lca;
};
node edge[N*2];
void add(int u,int v,int w){//存图
edge[cnt].w=w;
edge[cnt].to=v;
edge[cnt].next=head[u];
head[u]=cnt++;
}
node qedge[N*2];
void qadd(int u,int v){//存询问的边
qedge[qcnt].to=v;
qedge[qcnt].next=qhead[u];
qhead[u]=qcnt++;
}
void init(){//初始化
cnt=0;qcnt=0;
memset(head,-1,sizeof(head));
memset(qhead,-1,sizeof(qhead));
memset(fa,0,sizeof(fa));
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
}
int find(int x){//找父节点
if(fa[x]==x) return x;
else return fa[x]=find(fa[x]);
}
void tarjan(int u){//把整棵树的一部分看作以节点x为根节点的小树
fa[u]=u;//自己可以是自己的父节点
vis[u]=1;//标记
for(int i=head[u];i!=-1;i=edge[i].next){//遍历所有与x相邻的节点
int v = edge[i].to;
if(!vis[v]){
dis[v]=dis[u]+edge[i].w;
tarjan(v);//以该节点为根节点建小树
fa[v]=u;//归为一类
}
}
for(int i=qhead[u];i!=-1;i=qedge[i].next){//搜寻包含x的所有询问
int v = qedge[i].to;
if(vis[v]){//如果 另一节点已被搜索过
qedge[i].lca=find(v);//把另一节点的父节点设为最近公共祖先
// cout<<v<<" "<<find(v)<<endl;
qedge[i].dis=dis[u]+dis[v]-2*dis[find(v)];//两节点之间的距离
qedge[i^1].lca=qedge[i].lca;
qedge[i^1].dis=qedge[i].dis;//一个公共祖先节点有两个子节点,两个点的dis相等
}
}
}
int main(){
int u,v,w;
while(~scanf("%d %d",&n,&m)){
init();
char ch;
//cout<<1<<endl;
for(int i=1;i<=m;i++){
scanf("%d %d %d %c",&u,&v,&w,&ch);//ch并没有什么用
add(u,v,w);
add(v,u,w);
}
cin>>w;
for(int i=1;i<=w;i++){
cin>>u>>v;
qadd(u,v);
qadd(v,u);
}
// for(int i=qhead[1];i!=-1;i=qedge[i].next){
// cout<<qedge[i].to<<endl;
// break;
// }
tarjan(1);
for(int i=0;i<qcnt;i+=2){//输出,注意i+=2
cout<<qedge[i].dis<<endl;
}
}
return 0;
}