poj 1986tarjan模板题

#include<iostream>

#include<vector>

using namespace std;

const int N=40010;

int pre[N];//并查集

int visit[N];//是否经过

int ancestor[N];//祖先

int dis[N];//储存距离

int result[N];//储存结果

struct tre{

 int x,length;

};

vector<tre>tree[N];

struct  qu{

 int y,index;

};

vector<qu>qury[N];

void init(int n) {//初始化

 int i;

 for(i=1;i<=n;i++)

  tree[i].clear();

 for(i=1;i<=n;i++)

  qury[i].clear();

 memset(visit,0,sizeof(visit));

}

int find(int x) {//路径压缩

 if(x!=pre[x])

  pre[x]=find(pre[x]);

 return pre[x];

}

void infind(int x,int y) {//合并

 int f1=find(x);

 int f2=find(y);

    pre[f1]=f2;

 return ;

}

void tarjan(int u,int length) {

 visit[u]=1;

 dis[u]=length;//直接就可以的到距离

 ancestor[u]=u;

       pre[u]=u;

 int i;

 for(i=0;i<tree[u].size();i++) {

  int v=tree[u][i].x;

  if(visit[v]==0) {

   tarjan(v,length+tree[u][i].length);

   infind(u,v);

   ancestor[find(u)]=u;

  }

 }

  for(i=0;i<qury[u].size();i++) {

   int v=qury[u][i].y;

   if(visit[v])

    result[qury[u][i].index]=dis[u]+dis[v]-2*dis[ancestor[find(v)]];//重点

  }

  return ;

}

int main() {

 int n,m,i,j,k,a,b,c;

 char s[2];

 while(scanf("%d%d",&n,&m)!=EOF) {

  init(n);

  for(i=1;i<=m;i++) {

   scanf("%d%d%d%s",&a,&b,&c,s);

   tre h;

   h.x=b;

   h.length=c;

   tree[a].push_back(h);//储存

   h.x=a;

   h.length=c;

   tree[b].push_back(h);

  }

  scanf("%d",&k);

  for(i=1;i<=k;i++) {

   scanf("%d%d",&a,&b);

   qu h;

    h.y=b;

    h.index=i;

    qury[a].push_back(h);//储存

    h.y=a;

    h.index=i;

    qury[b].push_back(h);

  }

  tarjan(1,0);//

  for(i=1;i<=k;i++)

   printf("%d\n",result[i]);

 }

 return 0;

}