Network POJ - 3417 lca+ dfs

参考http://blog.csdn.net/l123012013048/article/details/47738691
照着别人的代码，

1. 这个地方的用数组实现队列挺好的这样，就可以从叶子开始到根了
2. 这个地方的 小技巧应该比较普遍吧，dp【u】++，dp【v】++，dp【lca】-=2；

https://vjudge.net/contest/51670#rank

int tot,fst[N],vv[N<<2],nxt[N<<2];
int dep[N],dp[N];
int fa[22][N];
int n,m;

void add(int u,int v ){
    vv[tot]=v,nxt[tot]=fst[u];fst[u]=tot++;
}
void init(){
    mem(fst,-1);tot=0;mem(dep,0);mem(dp,0);
}
int lca(int u,int v){
    if(dep[u]>dep[v])swap(u,v);
    for(int k=19;k>=0;--k){
        if((dep[v]-dep[u])>>k&1)v=fa[k][v];
    }
    if(u==v)return u;
    for(int k=20;k>=0;--k){
        if(fa[k][u]!=fa[k][v])
            u=fa[k][u], v=fa[k][v];
    }
    return fa[0][u];
}
int q[N];
void lca_init(){
    dep[0]=0;
    int head=1,tail=1;
    q[1]=1;
    fa[0][1]=-1;
    while(head<=tail){
        int u=q[head++];
        for(int i=fst[u];~i;i=nxt[i]){
            int v=vv[i];
            if(v==fa[0][u])continue;
            fa[0][v]=u;
            dep[v]=dep[u]+1;
            q[++tail]=v;
        }
    }
    for(int k=0;k<20;++k){
        for(int i=1;i<=n;++i){
            if(fa[k][i]==-1)fa[k+1][i]=-1;
            else fa[k+1][i]=fa[k][fa[k][i]];
        }
    }
}
void read(){
    init();
    int u,v;
    rep(i,1,n-1)sf("%d%d",&u,&v),add(u,v),add(v,u);
    lca_init();
    rep(i,1,m){sf("%d%d",&u,&v);dp[u]++;dp[v]++;dp[lca(u,v)]-=2;}
}
int main(){
    //freopen("in.txt","r",stdin);
    while(~sf("%d%d",&n,&m)){
        read();
        LL ans=0;
        for(int i=n;i>=1;--i){
            int v=q[i];
            dp[fa[0][v]]+=dp[v];
        }
        for(int i=2;i<=n;++i){
            if(dp[i]==0)ans+=m;
            else if(dp[i]==1)ans++;
        }
        cout<<ans<<'\n';
    }
}