题意
有一棵n个点的树和m条路径,问存在多少个点对(x,y)满足这两个点被至少一条路径同时包含。
n
,
m
≤
1
0
5
n,m\le10^5
n,m≤105
分析
枚举点x,那么能和x构成合法点对的点y的数量即为所有包含x的路径的并的大小,也就是所有满足路径的端点构成的虚树大小。
显然若我们把虚树的端点按照dfs序环形排列,那么虚树大小即为相邻两点距离和的一半。
那么我们可以用线段树来维护dfs序,然后用线段树合并来统计答案。
具体来讲就是把每条链在其两个端点处插入,在lca处删除,然后动态维护相邻点对的距离和即可。
时间复杂度
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define pb push_back
typedef long long LL;
const int N=100005;
int n,m,cnt,last[N],dfn[N],dep[N],tim,fa[N],app[N],eu[N*2],tot,lg[N*2],rmq[N*2][25],bin[25],rt[N],sz;
struct tree{int l,r,lef,rig,s;LL w;}t[N*50];
std::vector<int> vec1[N],vec2[N];
struct edge{int to,next;}e[N*2];
LL ans;
void addedge(int u,int v)
{
e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}
void dfs(int x)
{
dfn[x]=++tim;
eu[++tot]=x;
dep[x]=dep[fa[x]]+1;
app[x]=tot;
for (int i=last[x];i;i=e[i].next)
{
if (e[i].to==fa[x]) continue;
fa[e[i].to]=x;
dfs(e[i].to);
eu[++tot]=x;
}
}
void get_rmq()
{
for (int i=1;i<=tot;i++) lg[i]=log(i)/log(2),rmq[i][0]=eu[i];
bin[0]=1;
for (int i=1;i<=lg[tot];i++) bin[i]=bin[i-1]*2;
for (int j=1;j<=lg[tot];j++)
for (int i=1;i+bin[j]-1<=tot;i++)
rmq[i][j]=dep[rmq[i][j-1]]<dep[rmq[i+bin[j-1]][j-1]]?rmq[i][j-1]:rmq[i+bin[j-1]][j-1];
}
int get_lca(int x,int y)
{
x=app[x];y=app[y];
if (x>y) std::swap(x,y);
int k=lg[y-x+1];
return dep[rmq[x][k]]<dep[rmq[y-bin[k]+1][k]]?rmq[x][k]:rmq[y-bin[k]+1][k];
}
void updata(int d)
{
t[d].w=t[t[d].l].w+t[t[d].r].w;
t[d].lef=t[t[d].l].lef?t[t[d].l].lef:t[t[d].r].lef;
t[d].rig=t[t[d].r].rig?t[t[d].r].rig:t[t[d].l].rig;
int p=t[t[d].l].rig,q=t[t[d].r].lef;
if (p&&q) t[d].w+=dep[p]+dep[q]-2*dep[get_lca(p,q)];
}
void ins(int &d,int l,int r,int x,int y)
{
if (!d) d=++sz;
if (l==r)
{
if (!y) t[d].s-=2,t[d].lef=t[d].rig=!t[d].s?0:t[d].lef;
else t[d].s++,t[d].lef=t[d].rig=y;
return;
}
int mid=(l+r)/2;
if (x<=mid) ins(t[d].l,l,mid,x,y);
else ins(t[d].r,mid+1,r,x,y);
updata(d);
}
int merge(int x,int y,int l,int r)
{
if (!x||!y) return x^y;
if (l==r) {t[x].s+=t[y].s;return x;}
int mid=(l+r)/2;
t[x].l=merge(t[x].l,t[y].l,l,mid);
t[x].r=merge(t[x].r,t[y].r,mid+1,r);
updata(x);
return x;
}
void solve(int x)
{
for (int i=last[x];i;i=e[i].next)
{
if (e[i].to==fa[x]) continue;
solve(e[i].to);
rt[x]=merge(rt[x],rt[e[i].to],1,n);
}
for (int i=0;i<vec1[x].size();i++) ins(rt[x],1,n,dfn[vec1[x][i]],vec1[x][i]);
ans+=t[rt[x]].w;
int p=t[rt[x]].lef,q=t[rt[x]].rig;
if (p&&q) ans+=dep[p]+dep[q]-2*dep[get_lca(p,q)];
for (int i=0;i<vec2[x].size();i++) ins(rt[x],1,n,dfn[vec2[x][i]],0);
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<n;i++)
{
int x,y;scanf("%d%d",&x,&y);
addedge(x,y);
}
dfs(1);
get_rmq();
for (int i=1;i<=m;i++)
{
int x,y;scanf("%d%d",&x,&y);
int lca=get_lca(x,y);
vec1[x].pb(x);vec1[x].pb(y);vec1[y].pb(x);vec1[y].pb(y);
vec2[lca].pb(x);vec2[lca].pb(y);
}
solve(1);
printf("%lld\n",ans/4);
return 0;
}