比赛 20241128 评测结果 TMMMTTMMMMMMTTMMMMTM
题目名称 猴猴的比赛 最终得分 0
用户昵称 flyfree 运行时间 29.256 s
代码语言 C++ 内存使用 364.71 MiB
提交时间 2024-11-28 09:21:29
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 100010
inline ll read(){
	ll x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		x=x*10+c-'0';
		c=getchar();
	}
	return x*f;
}
struct node{
	ll l[MAXN*40],r[MAXN*40],s[MAXN*40][2],siz[MAXN*40];
	ll rub[MAXN],id,idx;
	ll newnode(ll lz,ll rz){
		ll now; 
		if(id){
			now=rub[id--];
		}else now=++idx;
		s[now][0]=s[now][1]=0;
		l[now]=lz,r[now]=rz;
		siz[now]=1;
		return now;
	}
//	void push_up(ll now){
//		siz[now]=siz[s[now][1]]+siz[s[now][0]];
//	}
	void build(ll &now,ll pos,ll lz,ll rz){
		now=newnode(lz,rz);
//		cout<<now<<" lz:"<<lz<<" rz:"<<rz<<endl;
		if(lz==rz)return;
		ll mid=(lz+rz)/2;
		if(pos<=mid)build(s[now][0],pos,lz,mid);
		else build(s[now][1],pos,mid+1,rz);
	}
	void merge(ll &p,ll q){
		if((!p)||(!q)){
			p=p|q;
			if(q)rub[++id]=q;
			return;
		}
		siz[p]+=siz[q];
		merge(s[p][0],s[q][0]);
		merge(s[p][1],s[q][1]);
//		push_up(p);
	}
	ll find(ll now,ll lz,ll rz){
		if(!now)return 0;
//		cout<<l[now]<<" "<<r[now]<<" "<<siz[now]<<endl;
		if(l[now]>=lz&&r[now]<=rz)return siz[now];
		ll mid=(l[now]+r[now])/2,ans=0;
		if(lz<=mid)ans+=find(s[now][0],lz,rz);
		if(rz>mid)ans+=find(s[now][1],lz,rz);
		return ans;
	}
};
node tr;
ll n,idx,cnt,ans;
ll hd[MAXN*2],nxt[MAXN*4],ed[MAXN*4];
ll dfn[MAXN],lst[MAXN],rot[MAXN];
void build(ll x,ll y){
	nxt[++idx]=hd[x];
	hd[x]=idx;
	ed[idx]=y;
}
void dfs(ll now,ll fa){
	dfn[now]=lst[now]=++cnt;
	for(int i=hd[now];i;i=nxt[i]){
		ll y=ed[i];
		if(y==fa)continue;
		dfs(y,now);
		lst[now]=max(lst[now],lst[y]);
	}
}
void dfs2(ll now,ll fa){
	tr.build(rot[now-n],dfn[now-n],1,n);
//	cout<<rot[now-n]<<endl;
	for(int i=hd[now];i;i=nxt[i]){
		ll y=ed[i];
		if(y==fa)continue;
		dfs2(y,now);
		tr.merge(rot[now-n],rot[y-n]);
	}
	ans+=tr.find(rot[now-n],dfn[now-n],lst[now-n])-1;
//	cout<<now-n<<" "<<dfn[now-n]<<" "<<lst[now-n]<<" "<<ans<<endl;
}
int main(){
	freopen("monkeyclim.in","r",stdin);
	freopen("monkeyclim.out","w",stdout);
	n=read();
	for(int i=1;i<n;i++){
		ll x=read(),y=read();
		build(x,y);
		build(y,x);
	}
	for(int i=1;i<n;i++){
		ll x=read()+n,y=read()+n;
		build(x,y);
		build(y,x);
	}
	dfs(1,0);
	dfs2(1+n,0);
	cout<<ans;
	return 0;
}