ZROI2018提高day5t2

传送门

分析

考场上傻了，写了个树剖还莫名weila......

实际就是按顺序考虑每个点，然后从他往上找，一边走一边将走过的边染色，如果走到以前染过色的边就停下。对于每一个a[i]的答案就是之前走过的所有边的数量*2-它自己的深度。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
int col[],sum,fa[],dep[];
vector<int>v[];
inline void dfs(int x){
      for(int i=;i<v[x].size();i++)
        if(v[x][i]!=fa[x]){
          fa[v[x][i]]=x;
          dep[v[x][i]]=dep[x]+;
          dfs(v[x][i]);
        }
      return;
}
int main(){
      int n,m,i,j,k;
      scanf("%d%d",&n,&m);
      for(i=;i<n;i++){
          int x,y;
          scanf("%d%d",&x,&y);
          v[x].push_back(y);
          v[y].push_back(x);
      }
      fa[]=;
      dep[]=;
      dfs();
      col[]=;
      for(i=;i<=m;i++){
          int x,y;
          scanf("%d",&x);
          y=x;
          while(!col[y]){
            col[y]=;
            sum++;
            y=fa[y];
          }
          printf("%d\n",sum*-dep[x]);
      }
      return ;
}