【bzoj3522】 [Poi2014]Hotel

Description

有一个树形结构的宾馆，n个房间，n-1条无向边，每条边的长度相同，任意两个房间可以相互到达。吉丽要给他的三个妹子各开（一个）房（间）。三个妹子住的房间要互不相同（否则要打起来了），为了让吉丽满意，你需要让三个房间两两距离相同。

有多少种方案能让吉丽满意？

Input

第一行一个数n。

接下来n-1行，每行两个数x,y，表示x和y之间有一条边相连。

Output

让吉丽满意的方案数。

Sample Input

7

1 2

5 7

2 5

2 3

5 6

4 5

Sample Output

5

HINT

【样例解释】

{1,3,5},{2,4,6},{2,4,7},{2,6,7},{4,6,7}

【数据范围】

n≤5000

先枚举一个点，然后dfs计数

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=5005;
int fi
,ne[N<<1],j[N<<1];
int n,mx;
long long t1
,t2
,s
,ans;
void dfs(int ro,int fa,int dep){
mx=max(mx,dep);s[dep]++;
for (int u=fi[ro];u;u=ne[u])
if (j[u]!=fa)dfs(j[u],ro,dep+1);
}
int main (){
scanf ("%d",&n);
for (int i=1;i<n;++i){
scanf ("%d%d",&j[i+n],&j[i]);
ne[i]=fi[j[i+n]];fi[j[i+n]]=i;
ne[i+n]=fi[j[i]];fi[j[i]]=i+n;
}
for (int i=1;i<=n;++i){
memset(t1,0,sizeof(t1));
memset(t2,0,sizeof(t2));
for (int u=fi[i];u;u=ne[u]){
dfs(j[u],i,1);
for (int k=1;k<=mx;++k){
ans+=t2[k]*s[k];
t2[k]+=t1[k]*s[k];
t1[k]+=s[k];
s[k]=0;
}
}
}
printf ("%lld",ans);
return 0;
}