codeforces 191C Fools and Roads (树链剖分 或 树上差分) HQG_AC 的博客

#include <bits/stdc++.h>
using namespace std ;
const int N = 100000 + 10 ;
struct node{
    int v,next,en ;
}e[2*N];
int cf[N] ;
int ans[N] ;
bool vis[N] ;
int tot,a,b,n,u,v,k ;
int head[N],deep[N] ;
int f[N][32] ;
void add_edge(int u,int v,int en)
{
    e[tot].en=en ;
    e[tot].v=v ;
    e[tot].next=head[u] ;
    head[u]=tot++;
}
//两个点的地方打个标记 lca的地方再打个标记 然后最后dfs一遍统计下标记
void dfs(int u,int fa) {
    deep[u]=deep[fa]+1 ;
    f[u][0]=fa ;//表示x向上走了2^0(=1)部到达的点 
    for (int i=1;(1<<i)<=deep[u];i++) f[u][i]=f[f[u][i-1]][i-1] ;
    /*因为在循环到i时f[x][i-1]一定已经求好，
    又因为f[x]的2^i祖先=f[x]的2^(i-1)祖先的2^(i-1)祖先，所以这么求 */
    for (int i=head[u];i!=-1;i=e[i].next) {
        int v=e[i].v ;
        if (v!=fa) dfs(v,u) ;
    }
}
int LCA(int a,int b){ //最近公共祖先 
    if (deep[a]>deep[b]) swap(a,b) ;
    for (int i=31;i>=0;i--) //把a和b放到同一深度 
        if (deep[a]<=deep[b]-(1<<i)) b=f[b][i] ;
    if (a==b) return a;
    for (int i=31;i>=0;i--){
        if (f[a][i]==f[b][i]) continue ;
        else a=f[a][i],b=f[b][i] ;
    } 
    return f[a][0] ;
} 
void make_cf(int u){
    vis[u]=true ;
    for (int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].v ;
        if (vis[v]) continue ; 
        make_cf(v) ;
        cf[u]+=cf[v] ;
        ans[e[i].en]+=cf[v] ;
    }
}
int main()
{
    memset(head,-1,sizeof(head)) ;

    scanf("%d",&n) ;
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v) ;
        add_edge(u,v,i) ;
        add_edge(v,u,i) ;
    }
    dfs(1,0) ;
    scanf("%d",&k) ;
    for (int i=1;i<=k;i++)
    {
        scanf("%d%d",&u,&v) ;
        cf[u]++;cf[v]++;cf[LCA(u,v)]-=2 ; 
    }
    memset(vis,false,sizeof(vis)) ;
    memset(ans,0,sizeof(ans)) ;
    make_cf(1) ;
    for (int i=1;i<n;i++) printf("%d ",ans[i]) ;
}