每日一题 P3128 [USACO15DEC]Max Flow P LCA倍增 lazy离线处理做法
又是一道树上倍增的绿题,思路依旧很简单。本题可以用树上差分来做,也非常方便,但是当时看到这题第一反应就是这种标记做法,于是就这么写出来了。
常规的求倍增lca就不说了,主要是每条路径上面所有的点的值都需要+1。倍增每次网上跳一段距离,这边我们借用lazy tag的思想,每次lca向上跳的时候把跳的那一段做一个标记,最后询问的时候统一下放到每个点,更新的方式和预处理倍增差不多,只不过反过来进行操作了,具体见代码。这样就可以在nlogn的时间内得到答案。
#include <bits/stdc++.h>
#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();
}