【題目描述】

【思路】
看了大佬的題解才想明白的，f_zyj大佬的題解
兩棵樹，對第一棵樹暴力列舉所有邊，拆掉這條邊後的兩個子樹對應兩個集合 $A1,B1$

#include<bits/stdc++.h>
#define max(a,b)(a>b?a:b)
using namespace std;
const int maxn=4005;

struct Edge{
	int from,to;
	Edge(int f=0,int t=0):from(f),to(t){}
};

int n,a1;
long long ans;
bool used[maxn];
int num[maxn],dp[maxn];
vector<int> g1[maxn],g2[maxn];
Edge edges[maxn];

void dfs1(int u,int fa){
	used[u]=true;
	++a1;
	for(int i=0;i<g1[u].size();++i){
		int v=g1[u][i];
		if(v!=fa && !used[v]) dfs1(v,u);
	}
}

void dfs2(int u,int fa){
	num[u]=1;
	dp[u]=used[u]?1:0;
	for(int i=0;i<g2[u].size();++i){
		int v=g2[u][i];
		if(v!=fa){
			dfs2(v,u);
			num[u]+=num[v];
			dp[u]+=dp[v];
		}
	}
	if(u!=0){//u=0時樹沒有被分成兩部分所以不算 
		int b1=num[u];
		//集合a1和b1的交集大小為dp[u]
		int maxv=0;
		maxv=max(maxv,dp[u]);
		maxv=max(maxv,a1-dp[u]);
		maxv=max(maxv,b1-dp[u]);
		maxv=max(maxv,n-a1-b1+dp[u]);
		ans+=(long long)maxv*maxv;
	}
}

int main(){
	scanf("%d",&n);
	for(int i=0;i<n-1;++i){
		int u,v;
		scanf("%d%d",&u,&v);
		g1[u].push_back(v);
		g1[v].push_back(u);
		edges[i]=Edge(u,v);
	}
	for(int i=0;i<n-1;++i){
		int u,v;
		scanf("%d%d",&u,&v);
		g2[u].push_back(v);
		g2[v].push_back(u);
	}
	for(int i=0;i<n-1;++i){
		a1=0;
		memset(used,0,sizeof(used));
		dfs1(edges[i].from,edges[i].to);
		dfs2(0,-1);
	}
	printf("%lld\n",ans);
	return 0;
}