LG3565 [POI2014]HOT-Hotels

题意

有一个树形结构，每条边的长度相同，任意两个节点可以相互到达。选3个点。两两距离相等。有多少种方案？

1≤n≤5 000

分析

参照小塘空明的题解。

很明显到一个点距离相等的三个点两两之间距离相等。

所以我们枚举该点，对子树进行暴力统计,注意统计的顺序

时间复杂度\(O(n^2)\)

代码

#include<bits/stdc++.h>
#define rg register
#define il inline
#define co const
template<class T>il T read(){
	rg T data=0,w=1;
	rg char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') w=-1;
		ch=getchar();
	}
	while(isdigit(ch))
		data=data*10+ch-'0',ch=getchar();
	return data*w;
}
template<class T>il T read(rg T&x){
	return x=read<T>();
}
typedef long long ll;

co ll size=5e3+1;
ll n,ans,tot,mx,d[size],tmp[size],s1[size],s2[size];
ll head[size],ver[size*2],next[size*2];
void add(ll x,ll y){
	ver[++tot]=y,next[tot]=head[x],head[x]=tot;
}
void dfs(ll x,ll fa){
	mx=std::max(mx,d[x]);
	tmp[d[x]]++;
	for(ll i=head[x];i;i=next[i]){
		ll y=ver[i];
		if(y==fa) continue;
		d[y]=d[x]+1,dfs(y,x);
	}
}
int main(){
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	read(n);
	for(ll i=1;i<n;++i){
		ll x=read<ll>(),y=read<ll>();
		add(x,y),add(y,x);
	}
	for(ll x=1;x<=n;++x){
		memset(s1,0,sizeof s1);
		memset(s2,0,sizeof s2);
		for(ll i=head[x];i;i=next[i]){
			ll y=ver[i];
			mx=0,d[y]=1,dfs(y,x);
			for(ll j=1;j<=mx;++j){
				ans+=s2[j]*tmp[j];
				s2[j]+=s1[j]*tmp[j];
				s1[j]+=tmp[j];
			}
			for(ll j=1;j<=mx;++j) tmp[j]=0;
		}
	}
	printf("%lld\n",ans);
	return 0;
}