记录编号 597487 评测结果 AAAAAAAAAAAAEEEEEEEE
题目名称 猴猴的比赛 最终得分 60
用户昵称 Gravatarflyfree 是否通过 未通过
代码语言 C++ 运行时间 2.215 s
提交时间 2024-11-28 14:26:00 内存使用 7.39 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 10010
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;
}
bitset <MAXN> vec[MAXN];
ll hd[MAXN*2],nxt[MAXN*4],ed[MAXN*4];
ll n,idx,ans;
void build(ll x,ll y){
	nxt[++idx]=hd[x];
	ed[idx]=y;
	hd[x]=idx;
}
void dfs1(ll now,bitset<MAXN> s){
	vec[now]=s;
//	cout<<now<<" ";
//	for(int i=1;i<=n;i++)cout<<s[i];
//	cout<<endl;
	s[now]=true;
//	cout<<now<<" "<<s<<endl;
	for(int i=hd[now];i;i=nxt[i]){
		ll y=ed[i];
		if(s[y])continue;
		dfs1(y,s);	
	}
}
void dfs2(ll now,bitset<MAXN> s){
	bitset <MAXN> p=s&vec[now-n];
	ans+=p.count();
//	cout<<now<<" ";
//	for(int i=1;i<=n;i++)cout<<s[i];
//	cout<<" ";
//	for(int i=1;i<=n;i++)cout<<p[i];
//	cout<<endl;
	s[now-n]=true;
	for(int i=hd[now];i;i=nxt[i]){
		ll y=ed[i];
		if(s[y-n])continue;
		dfs2(y,s);
	}
}
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);
	}
	dfs1(1,0);
	dfs2(n+1,0);
	cout<<ans;
	return 0;
}