【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

题解

枚举三个点中心开始DP

代码（by hzwer）

#include<bits/stdc++.h>
#define inf 1000000000
#define ll long long
using namespace std;
ll read()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,mx,cnt;
int last[5005],deep[5005],tmp[5005];
ll s1[5005],s2[5005],ans;
struct edge{
int to,next;
}e[10005];
void insert(int u,int v)
{
e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}
void dfs(int x,int fa)
{
mx=max(deep[x],mx);
tmp[deep[x]]++;
for(int i=last[x];i;i=e[i].next)
if(e[i].to!=fa)
{
deep[e[i].to]=deep[x]+1;
dfs(e[i].to,x);
}
}
int main()
{
n=read();
for(int i=1;i<n;i++)
{
int u=read(),v=read();
insert(u,v);
}
for(int x=1;x<=n;x++)
{
memset(s1,0,sizeof(s1));
memset(s2,0,sizeof(s2));
for(int i=last[x];i;i=e[i].next)
{
deep[e[i].to]=1;
dfs(e[i].to,x);
for(int j=1;j<=mx;j++)
{
ans+=s2[j]*tmp[j];
s2[j]+=tmp[j]*s1[j];
s1[j]+=tmp[j];
}
for(int j=1;j<=mx;j++)
tmp[j]=0;
}
}
printf("%lld\n",ans);
return 0;
}