每日一题 P3128 [USACO15DEC]Max Flow P LCA倍增 lazy离线处理做法

#include 
#define MAXN 500005
#define int long long
using namespace std;
struct EDGE
{
    int to,next;
} edge[MAXN];
int head[MAXN],ptr;
void add_edge(int u,int v)
{
    edge[++ptr].to=v;
    edge[ptr].next=head[u];
    head[u]=ptr;
}
void add(int u,int v)
{
    add_edge(u,v);
    add_edge(v,u);
}
int dep[MAXN],f[MAXN][21],cnt[MAXN][21];
int n,k;
void dfs(int now,int fa)
{
    for(int p=head[now]; p; p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==fa) continue;
        dep[to]=dep[now]+1;//计算深度
        f[to][0]=now;
        dfs(to,now);
    }
}
void dp()
{
//预处理 f[u][i] u结点往上走2^i次
    for(int i=1; (1<<i)<=n; i++)
        for(int u=1; u<=n; u++)
            f[u][i]=f[f[u][i-1]][i-1];//u结点往上走2^i次等于网上走两个2^(i-1)
}
void lca(int x,int y)
{
    int p,t;
    if(dep[x]<dep[y])
        swap(x,y);//保证x比y大
    for(p=0; (1<<p)<=dep[x]; p++);//算出dp最多往上走2的多少次
    for(t=--p; t>=0; t--)//从x走到y同一层
        if(dep[x]-(1<<t)>=dep[y])//不超过的话就往上走
            cnt[x][t]++,x=f[x][t];
    if(x==y)
    {
        cnt[x][0]++;
        return;
    }
    for(t=p; t>=0; t--)//x和y一起走 不相等就往上走
        if(f[x][t]!=f[y][t])
            cnt[x][t]++,x=f[x][t],cnt[y][t]++,y=f[y][t];
    cnt[x][1]++,cnt[y][0]++;
}
int cal()
{
    for(int i=20; i>=1; i--)
        for(int u=n; u>=1; u--)
            cnt[u][i-1]+=cnt[u][i],cnt[f[u][i-1]][i-1]+=cnt[u][i];
    int ret=0;
    for(int i=1;i<=n;i++)
        ret=max(ret,cnt[i][0]);
    return ret;
}
signed main()
{
    cin>>n>>k;
    for(int i=1;i<n;i++)
    {
        int a,b;cin>>a>>b;
        add(a,b);
    }
    dfs(1,0);dp();
    for(int i=1;i<=k;i++)
    {
        int a,b;cin>>a>>b;
        lca(a,b);
    }
    cout<<cal();
}