1471. Tree(LCA)

1471

先学习了下tarjan算法的LCA 离线算法 它是先知道询问的结点对 在遍历的时候就已经算出来了

看篇图解 讲的很清楚

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<stdlib.h>
 using namespace std;
 #define N 100010
 struct node
 {
     int u,v,w,next;
 }ed[N<<];
 int head[N],t,n,q,dis[N];
 int vis[N],ans[N],qhead[N<<];
 int father[N],x[N],y[N];
 void init()
 {
     t = ;
     memset(head,-,sizeof(head));
     memset(qhead,-,sizeof(qhead));
 }
 void add(int u,int v,int w)
 {
     ed[t].u = u;
     ed[t].v = v;
     ed[t].w = w;
     ed[t].next = head[u];
     head[u] = t++;
 }
 void add1(int u,int v,int d)
 {
     ed[t].w = d;
     ed[t].u = u;
     ed[t].v = v;
     ed[t].next = qhead[u];
     qhead[u] = t++;
 }
 int find(int x)
 {
     if(x!=father[x])
     father[x] = find(father[x]);
     return father[x];
 }
 void tarjan(int u)
 {
     int i;
     for(i = qhead[u] ; i!=- ; i = ed[i].next)
     {
         int v = ed[i].v;
         if(vis[v])
         ans[ed[i].w] = find(v);
     }
     for(i = head[u] ; i!=- ; i = ed[i].next)
     {
         int v = ed[i].v;
         if(!vis[v])
         {
            vis[v] = ;
            dis[v] = dis[u]+ed[i].w;
            tarjan(v);
            father[v] = u;
         }
     }
 }
 int main()
 {
     int i,a,b,c;
     init();
     scanf("%d",&n);
     for(i = ; i < n ; i++)
     father[i] = i;
     for(i =  ; i < n ; i++)
     {
         scanf("%d%d%d",&a,&b,&c);
         add(a,b,c);
         add(b,a,c);
     }
     scanf("%d",&q);
     for(i = ; i <= q ; i++)
     {
         scanf("%d%d",&x[i],&y[i]);
         add1(x[i],y[i],i);
         add1(y[i],x[i],i);
     }
     vis[] = ;
     tarjan();
     for(i = ; i <= q ; i++)
     {
         printf("%d\n",dis[x[i]]+dis[y[i]]-*(dis[ans[i]]));
     }
     return ;
 }