CodeForces191 C. Fools and Roads (LCA+树上差分)

题意：

给一颗n个点的树
m次操作，每次操作给两个点a和b，将a到b路径上的每条边权值加1
问m次操作之后每条边的权值是多少

解法：

树上差分的裸题

先转化为有根树
假设要修改a,b，令lca(a,b)=lc
那么val(a)++,val(b)++,val(lc)-=2

最后自底向上传递一下值就行了

为什么不是val(lc)++,val(a)–,val(b)–呢？
因为一个节点对应多个子节点，向下传递的时候可能会传递到无关节点，手画一下样例就懂了
而每个节点的父节点是唯一的，因此向上传递不会错。

code：

#include<bits/stdc++.h>
using namespace std;
const int maxm=1e5+5;
int head[maxm],nt[maxm<<1],to[maxm<<1],w[maxm<<1],cnt;
int f[maxm][25],d[maxm];
const int maxd=20;
int ans[maxm];
int val[maxm];
void add(int x,int y,int z){
    cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;w[cnt]=z;
}
void dfs(int x,int fa){
    for(int i=head[x];i;i=nt[i]){
        int v=to[i];
        if(v==fa)continue;
        d[v]=d[x]+1;
        f[v][0]=x;
        dfs(v,x);
    }
}
int lca(int a,int b){
    if(d[a]<d[b])swap(a,b);
    for(int i=maxd;i>=0;i--){
        if(d[f[a][i]]>=d[b]){
            a=f[a][i];
        }
    }
    if(a==b)return a;
    for(int i=maxd;i>=0;i--){
        if(f[a][i]!=f[b][i]){
            a=f[a][i],b=f[b][i];
        }
    }
    return f[a][0];
}
void cal(int x,int fa){
    for(int i=head[x];i;i=nt[i]){
        int v=to[i];
        if(v==fa)continue;
        cal(v,x);
        ans[w[i]]+=val[v];
        val[x]+=val[v];
    }
}
signed main(){
    int n;scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        add(a,b,i);
        add(b,a,i);
    }
    dfs(1,-1);
    f[1][0]=1;
    for(int j=1;j<=maxd;j++){
        for(int i=1;i<=n;i++){
            f[i][j]=f[f[i][j-1]][j-1];
        }
    }
    int k;scanf("%d",&k);
    while(k--){
        int a,b;scanf("%d%d",&a,&b);
        int lc=lca(a,b);
        val[a]++,val[b]++;
        val[lc]-=2;
    }
    cal(1,-1);
    for(int i=1;i<n;i++){
        printf("%d ",ans[i]);
    }
    return 0;
}

---