51Nod 1681 公共祖先 [树状数组做法]

有一个庞大的家族，共n人。已知这n个人的祖辈关系正好形成树形结构(即父亲向儿子连边)。

在另一个未知的平行宇宙，这n人的祖辈关系仍然是树形结构，但他们相互之间的关系却完全不同了，原来的祖先可能变成了后代，后代变成的同辈……

两个人的亲密度定义为在这两个平行宇宙有多少人一直是他们的公共祖先。

整个家族的亲密度定义为任意两个人亲密度的总和。

Input

第一行一个数n(1<=n<=100000)
接下来n-1行每行两个数x,y表示在第一个平行宇宙x是y的父亲。
接下来n-1行每行两个数x,y表示在第二个平行宇宙x是y的父亲。

Output

一个数，表示整个家族的亲密度。

Input示例

5
1 3
3 5
5 4
4 2
1 2
1 3
3 4
1 5

Output示例

6

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

树状数组+dfs序+思路~

我们可以用树状数组维护目前i位置之前的点数，那么我们遍历第一棵树，求出每个点的区间[l,r]，再遍历第二棵树，每次在遍历前记录下此时区间内的点数now=cal(ar[u])-cal(al[u]-1)，然后在遍历后回到该点时再求出这个点数now1，k=now1-now-1就是该点子树中的公共点，ans+=k*(k-1)/2即可。

#include<cstdio>
#include<cstring>
#define ll long long

int n,x,y,fi[100001],w[200001],ne[200001],cnt,in[100001],al[100001],ar[100001],root;
ll ans,c[100001];

int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}

void add(int u,int v)
{
	w[++cnt]=v;ne[cnt]=fi[u];fi[u]=cnt;
	w[++cnt]=u;ne[cnt]=fi[v];fi[v]=cnt;
}

void add1(int u)
{
	for(;u<=n;u+=u&(-u)) c[u]++;
}

int cal(int u)
{
	ll now=0;
	for(;u;u-=u&(-u)) now+=c[u];
	return now;
}

void dfs1(int u,int fa)
{
	al[u]=++cnt;
	for(int i=fi[u];i;i=ne[i]) if(w[i]!=fa) dfs1(w[i],u);
	ar[u]=cnt;
}

void dfs2(int u,int fa)
{
	ll now=cal(ar[u])-cal(al[u]-1);add1(al[u]);
	for(int i=fi[u];i;i=ne[i]) if(w[i]!=fa) dfs2(w[i],u);
	now=cal(ar[u])-cal(al[u]-1)-now-1;
	ans+=now*(now-1)/2;
}

int main()
{
	n=read();
	for(int i=1;i<n;i++) x=read(),y=read(),in[y]++,add(x,y);
	for(root=1;in[root];root++);cnt=0;dfs1(root,0);
	memset(fi,0,sizeof(fi));
	memset(in,0,sizeof(in));cnt=0;
	for(int i=1;i<n;i++) x=read(),y=read(),in[y]++,add(x,y);
	for(root=1;in[root];root++);cnt=0;dfs2(root,0);
	printf("%lld\n",ans);
	return 0;
}