题解CF191C Fools and Roads

题面：CF191C Fools and Roads

树上差分模板题吧。唯一需要注意的也是值得讲的就是：边化点。

对于(i,j)连接的边，我们记录这条边的方式就是把（i,j）中深度大的点的id作为这条边的编号，第二遍dfs统计的时候可以直接记下来。

提一下树上差分：

边上操作：++s[u],++s[v],s[lca(u,v)]-=2;

点上操作：++s[u],++s[v],--s[lca(u,v)],--s[fa[lca(u,v)]];

#include<cstdio>
#define it register int
#define il inline
const int N=500005;
int u,v,h[N],nxt[N],adj[N],t,n,fa[N][20],d[N],lc,uu[N],vv[N],s[N],id[N],m,ans[N];
il void add(){
    nxt[++t]=h[u],h[u]=t,adj[t]=v,nxt[++t]=h[v],h[v]=t,adj[t]=u;
}
il void dfs(it x){
    d[x]=d[fa[x][0]]+1;
    for(it i=h[x];i;i=nxt[i])
        if(adj[i]!=fa[x][0])
            fa[adj[i]][0]=x,dfs(adj[i]);
}
il void DFS(it x){
    for(it i=h[x];i;i=nxt[i])
        if(adj[i]!=fa[x][0])
            DFS(adj[i]),s[x]+=s[adj[i]];
    ans[id[x]]=s[x];
}
il void sp(int &p,int &q){
    p+=q,q=p-q,p-=q;
}
il void lca(it p,it q){
    if(d[p]<d[q]) sp(p,q);
    it ds=d[p]-d[q];
    for(it i=17;i>=0;--i)
        if(ds&(1<<i)) p=fa[p][i];
    if(p==q){lc=p;return;}
    for(it i=17;i>=0;--i)
        if(fa[p][i]!=fa[q][i])
            p=fa[p][i],q=fa[q][i];
    lc=fa[p][0];
}
il void fr(int &num){
    num=0;char c=getchar();int p=1;
    while(c<'0'||c>'9') c=='-'?p=-1,c=getchar():c=getchar();
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    num*=p;
}
int main(){
    fr(n);
    for(it i=1;i<n;++i) fr(u),fr(v),add(),uu[i]=u,vv[i]=v;
    dfs(1);
    for(it i=1;i<n;++i) d[uu[i]]>d[vv[i]]?id[uu[i]]=i:id[vv[i]]=i;
    fr(m);
    for(it j=1;j<=17;++j)
        for(it i=1;i<=n;++i)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    while(m--)
        fr(u),fr(v),++s[u],++s[v],lca(u,v),s[lc]-=2; 
    DFS(1);
    for(it i=1;i<n;++i) printf("%d ",ans[i]); 
    return 0;
}