问题:给定n条路径,求被经过最多的点被经过的次数。
设差分数组,那么对于任意一条路径
,我们可以这么更新
数组:
最终点被经过的次数(
):
AC代码:
#include<cstdio>
#include<iostream>
#define ri register int
using namespace std;
const int MAXN=100020;
int n,k,m,u[MAXN],v[MAXN],fst[MAXN],nxt[MAXN],xi,yi,ans;
int fa[MAXN][16],deep[MAXN],cnt[MAXN];
void dfs(int x,int father,int dep)
{
fa[x][0]=father,deep[x]=dep;
for(ri i=1;(1<<i)<=deep[x];i++)
fa[x][i]=fa[fa[x][i-1]][i-1];
for(ri k=fst[x];k>0;k=nxt[k])
if(v[k]!=father) dfs(v[k],x,dep+1);
}
int LCA(int x,int y)//求LCA
{
if(deep[x]<deep[y]) swap(x,y);
int step=deep[x]-deep[y];
for(ri i=15;i>=0;i--)
if(step&(1<<i)) x=fa[x][i];
if(x==y) return x;
for(ri i=15;i>=0;i--)
{
if(fa[x][i]!=fa[y][i])
{
x=fa[x][i];
y=fa[y][i];
}
}
return fa[x][0];
}
void dfs2(int x)//累加数组cnt的值,最终统计出来的cnt[i]即为点i被经过的次数。
{
for(ri k=fst[x];k>0;k=nxt[k])
if(v[k]!=fa[x][0])
{
dfs2(v[k]);
cnt[x]+=cnt[v[k]];
}
ans=max(ans,cnt[x]);
}
int main()
{
scanf("%d%d",&n,&k);
m=(n-1)<<1;
for(ri i=1;i<=m;i+=2)
{
scanf("%d%d",&u[i],&v[i]);
nxt[i]=fst[u[i]],fst[u[i]]=i;
u[i+1]=v[i],v[i+1]=u[i];
nxt[i+1]=fst[u[i+1]],fst[u[i+1]]=i+1;
}
dfs(1,0,0);
for(ri i=1;i<=k;i++)
{
scanf("%d%d",&xi,&yi);
int now=LCA(xi,yi);
cnt[xi]+=1,cnt[yi]+=1,cnt[fa[now][0]]-=1,cnt[now]-=1;
}
dfs2(1);
cout<<ans;
return 0;
}