LibreOJ #3046.「ZJOI2019」语言 线段树合并

题意

有一棵n个点的树和m条路径，问存在多少个点对(x,y)满足这两个点被至少一条路径同时包含。
$n,m\le 10^5$

分析

枚举点x，那么能和x构成合法点对的点y的数量即为所有包含x的路径的并的大小，也就是所有满足路径的端点构成的虚树大小。
显然若我们把虚树的端点按照dfs序环形排列，那么虚树大小即为相邻两点距离和的一半。
那么我们可以用线段树来维护dfs序，然后用线段树合并来统计答案。
具体来讲就是把每条链在其两个端点处插入，在lca处删除，然后动态维护相邻点对的距离和即可。
时间复杂度$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;
}