zjoi2019 语言

好题好题。

我们先对一个结点 \(u\) 进行分析。

发现能对 \(u\) 产生贡献的所有结点可以构成一个联通分量。

只有经过 \(u\) 才会对 \(u\) 产生贡献。

而我们不可能将一条链上的所有点都扔到 \(u\) 上，这显然不现实，肯定是进行计算，算出来的，而且光一个结点都需要 \(O(n^2)\) 的内存，明显很没有前途。

考虑一条链的两端 \(l,r\) 挂在 \(u\) 上，类似一个虚树，考虑有 \(n\) 个叶子的时候我们是如何计算一颗虚树的大小的。

先对其的 dfs 序排序，\(siz = \sum dep_{a_i} - \sum dep_{LCA(a_i,a_i+1)} - dep _{LCA(a_1,a_n)}\)。dfs 序从小到大放，加入 dep 的时候会与 dfs 前面那一个产生交错， 这一部分就是 \(dep_{LCA(a_i,a_{i-1})}\) 。 最后我们因为钦定了 1 为根，去掉的话就是 \(-dep_{LCA_{i=1}^n a_i}\) 有dfs 序的性质可知其等于 \(-dep_{LCA(a_1,a_n)}\)。

好了那我们需要将所有的结点都算出来，明显可以树上差分。而我们发现如上的操作刚好可以使用线段树来维护，那树上的线段树肯定会想到线段树合并，不然这空间都得炸。

代码：

#include<bits/stdc++.h>
#define hnczy "language"
using namespace std;
const int N=5E5+5;
int st[20][N],dfn[N],lg[N];
int dep[N],fa[N],cnt,n,m;
vector<int>e[N],del[N];

long long ans;
int calc(int x,int y){return dep[x]<dep[y]?x:y;}
int LCA(int x,int y){
	if(!x || !y)return 0;
	if(x==y)return x;
	x=dfn[x],y=dfn[y];
	if(x>y)swap(x,y);
	
	int tmp=lg[y-x++];
	return calc(st[tmp][x],st[tmp][y-(1<<tmp)+1]);
} 
struct SEG{
	struct node{
		int ls,rs,L,R,val,c;
	}c[N<<4];
	#define ls(p) c[p].ls
	#define rs(p) c[p].rs
	#define mid ((l+r)>>1) 
	int rt[N],tot;
	void pushup(int p){
		c[p].L= c[ls(p)].L?c[ls(p)].L:c[rs(p)].L ;
	 	c[p].R= c[rs(p)].R?c[rs(p)].R:c[ls(p)].R ;
	 	c[p].val =c[ls(p)].val +c[rs(p)].val - dep[LCA(c[ls(p)].R,c[rs(p)].L )] ;
	}
	int query(int u){
		return c[u].val - dep[LCA(c[u].L,c[u].R)] ;
	}
	void change(int &p,int l,int r,int x,int w){
		if(!p)p = ++tot; 
		if(l==r){
			c[p].c +=w;
			c[p].val = (c[p].c ? dep[x]:0);
			c[p].L = c[p].R = (c[p].c ? x:0);
			return ;
		}
		if(dfn[x]<=mid)change(ls(p),l,mid,x,w);
		else change(rs(p),mid+1,r,x,w);
		pushup(p); 
	}
	void merge(int &p,int q,int l,int r){
		if(!p || !q)return p|=q,void();
		if(l==r)
			return c[p].c += c[q].c , c[p].val |= c[q].val, c[p].L |= c[q].L , c[p].R |= c[q].R,void() ;
		merge(ls(p),ls(q),l,mid),merge(rs(p),rs(q),mid+1,r);
		pushup(p);
	}
}seg;

void dfs1(int u,int F){
	dep[u] =dep[F] +1,fa[u] =F;
	st[0][dfn[u] = ++cnt]=F;
	for(int v:e[u])
		if(v!=F)dfs1(v,u);
}

void dfs2(int u){
	for(int v:e[u]) if(v!=fa[u])dfs2(v);
	for(int v:del[u]) seg.change(seg.rt[u],1,cnt,v,-1);
	ans+=seg.query(seg.rt[u]);seg.merge(seg.rt[fa[u]],seg.rt[u],1,cnt); 
}
int main(){
	freopen(hnczy ".in", "r", stdin);
    freopen(hnczy ".out", "w", stdout);
	scanf("%d%d",&n,&m);
	for(int i=1,u,v;i<n;i++){
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);  
	}
	dfs1(1,0);
	for(int i=1;i<=19;i++)
		for(int j=1;j+(1<<i-1)<=n;j++)
			st[i][j] = calc(st[i-1][j],st[i-1][j+(1<<i-1)]);
			
	for(int i=2;i<=n;i++)lg[i] = lg[i>>1]+1; 
	
	for(int i=1,u,v;i<=m;i++){
		scanf("%d%d",&u,&v);
		int lca= LCA(u,v);
		seg.change(seg.rt[u],1,cnt,u,1);
		
		seg.change(seg.rt[u],1,cnt,v,1);
		
		seg.change(seg.rt[v],1,cnt,u,1);
		
		seg.change(seg.rt[v],1,cnt,v,1);
		del[lca] .push_back(u),del[lca].push_back(v);
		del[fa[lca]] .push_back(u),del[fa[lca]].push_back(v);
	}
	dfs2(1);
	cout<<ans/2; 
	return 0;
}