计蒜客NOIP模拟赛4 D1T3 小X的佛光

小 X 是远近闻名的学佛，平日里最喜欢做的事就是蒸发学水。

小 X 所在的城市 X 城是一个含有 N 个节点的无向图，同时，由于 X 国是一个发展中国家，为了节约城市建设的经费，X 国首相在建造 X 城时只建造 N – 1 条边，使得城市的各个地点能够相互到达。

小 X 计划蒸发 Q 天的学水，每一天会有一名学水从 A 地走到 B 地，并在沿途各个地点留下一个水塘。此后，小 X 会从 C 地走到 B 地，并用佛光蒸发沿途的水塘。由于 X 城是一个学佛横行的城市，学水留下的水塘即使没有被小 X 蒸发，也会在第二天之前被其他学佛蒸发殆尽。

现在，小 X 想要知道，他每一天能够蒸发多少水塘呢?

输入格式

第一行三个整数 N,Q,num，分别表示 X 城地点的个数，小 X 蒸发学水的天数，以及测试点编号。注意，测试点编号是为了让选手们更方便的获得部分分，你可能不需要用到这则信息，在下发的样例中，测试点编号的含义是该样例满足某一测试点限制。

接下来 N – 1 行，每行两个整数 X,Y，表示 X 地与 Y 地之间有一条边。

接下来 Q 行，每行三个整数 A,B,C，表示一天中，有一名学水从 A 地走到 B 地，而小 X 会从 C 地走到 B 地。

输出格式

输出 Q 行，每行一个整数，表示小 X 能够蒸发的水塘数。

数据规模与约定

特殊性质 1：第 i 条边连接第 i 和第 i+1 个地点。

特殊性质 2：A=C。

样例输入

样例输出

1
1
3
A->B与B->C的共同路径长为
(LAB+LBC-LAC)/2
不过因为是点权，所以要+1
直接倍增LCA就行了
不过建树不能用dfs，会溢出，要用bfs()

 #include<iostream>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<queue>
 using namespace std;
 struct Node
 {
     int next,to;
 }edge[];
 int num,head[];
 int n,q,dep[],fa[][];
 bool vis[];
 void add(int u,int v)
 {
     num++;
     edge[num].next=head[u];
     head[u]=num;
     edge[num].to=v;
 }
 void bfs()
 {int i;
  queue<int>Q;
  memset(vis,,sizeof(vis));
   dep[]=;
  Q.push();
  vis[]=;
   while (Q.empty()==)
   {
       int u=Q.front();
       Q.pop();
       for (i=head[u];i;i=edge[i].next)
       {
           int v=edge[i].to;
           if (vis[v]==)
           {
               vis[v]=;
               dep[v]=dep[u]+;
               fa[v][]=u;
               Q.push(v);
           }
       }
   }
 }
 int LCA(int x,int y)
 {int i;
     if (dep[x]<dep[y]) swap(x,y);
     for (i=;i>=;i--)
         if ((<<i)<=dep[x]-dep[y]) x=fa[x][i];
  if (x==y) return x;
  while (x!=y)
  {
     for (i=;i>=;i--)
         if (fa[x][i]!=fa[y][i])
         {
             x=fa[x][i];
             y=fa[y][i];
         }
 
     if (x==y) return x;
     x=fa[x][];
     y=fa[y][];
  }
     return x;
 }
 int main()
 {int x,y,i,a,b,c,j;
     cin>>n>>q>>x;
     for (i=;i<=n-;i++)
     {
         scanf("%d%d",&x,&y);
         add(x,y);add(y,x);
     }
      bfs();
  for (i=;i<=n;i++)
     {
      for (j=;j<=;j++)
          fa[i][j]=fa[fa[i][j-]][j-];
     }
     for (i=;i<=q;i++)
     {
         scanf("%d%d%d",&a,&b,&c);
         int l1=LCA(a,b),l2=LCA(b,c),l3=LCA(a,c);
         int s1=dep[a]-dep[l1]+dep[b]-dep[l1];
         int s2=dep[b]-dep[l2]+dep[c]-dep[l2];
         int s3=dep[a]-dep[l3]+dep[c]-dep[l3];
         printf("%d\n",(s1+s2-s3)/+);
     }
 }