【bzoj3522】[Poi2014]Hotel 暴力+計數

題目大意：給定一棵樹，選取樹上三個點，使得三個點任意兩點之間距離相等，問方案數。

題目分析：對於任意三個點，要麼處於一條鏈，或者不處於同一條鏈。顯然處於一條鏈時不滿足題意。所以必然這三個點是有一箇中心使得該點到三個點距離相等。所以暴力枚舉中心，再計算方案數。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long LL;
struct info{
	int to,next;
}e[10005];

int first[5005],deep[5005],x,y,n,tot;
LL ans,t1[5005],t2[5005];
void add(int x,int y){
	e[++tot].to=y;
	e[tot].next=first[x];
	first[x]=tot;
}

void dfs(int u,int fa,int dep){
	deep[dep]++;
	for (int p=first[u];p;p=e[p].next){
		int v=e[p].to;
		if (fa!=v)
		dfs(v,u,dep+1);
	}
}


int main(){
	scanf("%d",&n);
	For (i,n-1){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	
	For (u,n){
		MEM(t1);
		MEM(t2);
		for (int p=first[u];p;p=e[p].next){
			MEM(deep);
			dfs(e[p].to,u,1);
			For (j,n){
				ans+=(LL)deep[j]*t2[j];
				t2[j]+=t1[j]*deep[j];
				t1[j]+=deep[j];
			}
			
		}
	}
	printf("%lld",ans);
	
}