P3128 Max Flow (LCA+树上差分)

题意：

给定一棵有N个点的树，所有节点的权值都为0。
有K次操作，每次指定两个点s,t，将s到t路径上所有点的权值都加一。
请输出K次操作完毕后权值最大的那个点的权值。

Sample Input

5 10
3 4
1 5
4 2
5 4
5 4
5 4
3 5
4 3
4 3
1 3
3 5
5 4
1 5
3 4

Sample Output

9

分析：

假设p[x]是点x的权值
u,v是要修改的两个点
lca(u,v)=z;
则p(u)++;p(v)++;p(z)- -;p(fa(z))- -;
为什么最后一个是fa(z)而不是z呢
因为z在u,v的路径中，需要加一次，所以z只需要减一次就行了

ps：
统计的时候开始想的从根开始从上到下统计，一直没搞出来怎么整
正解是从叶子向根从下到上统计

code:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxm=5e4+5;
int head[maxm],nt[maxm<<1],to[maxm<<1],cnt;
int f[maxm][20],maxd;
int d[maxm];
int p[maxm];
int sum[maxm];
int n,k;
int ans;
void init(){
    maxd=(int)(log(n)/log(2))+1;//最大倍增深度
    memset(head,-1,sizeof head);
    memset(d,0,sizeof d);
    memset(p,0,sizeof p);
    cnt=1;
    ans=0;
}
void add(int x,int y){
    cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;
}
void dfs(int x,int dep){
    d[x]=dep;
    for(int i=head[x];i!=-1;i=nt[i]){
        int v=to[i];
        if(!d[v]){
            f[v][0]=x;
            dfs(v,dep+1);
        }
    }
}
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){//统计
    ans=max(ans,p[x]);
    for(int i=head[x];i!=-1;i=nt[i]){
        int v=to[i];
        if(d[v]==d[x]+1){
            cal(v);
            p[x]+=p[v];
        }
    }
    ans=max(ans,p[x]);
}
int main(){
    while(scanf("%d%d",&n,&k)!=EOF){
        init();
        for(int i=1;i<=n-1;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            add(a,b);
            add(b,a);
        }
        f[1][0]=0;
        dfs(1,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];
            }
        }
        for(int i=1;i<=k;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            int t=lca(a,b);
            p[f[t][0]]--;
            p[t]--;
            p[a]++;
            p[b]++;
        }
        cal(1);
        printf("%d\n",ans);
    }
    return 0;
}