呵呵,一开始天真的我以为求个 西格玛 C(??,3)就好了。。
(题解:比枚举2个数的再多一个,,一样搞)
#include <bits/stdc++.h>
#define LL long long
#define lowbit(x) x&(-x)
#define inf 1e15
using namespace std;
inline int ra()
{
int x=,f=; char ch=getchar();
while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
return x*f;
}
struct edge{int next,to;}e[];
int tot[],mx,n,head[],cnt,s1[],s2[];
LL ans;
void insert(int x, int y){e[++cnt].next=head[x]; e[cnt].to=y; head[x]=cnt;}
void dfs(int x, int fa, int deep)
{
tot[deep]++;
mx=max(mx,deep);
for (int i=head[x];i;i=e[i].next)
{
if (e[i].to==fa) continue;
dfs(e[i].to,x,deep+);
}
}
int main(int argc, char const *argv[])
{
n=ra();
for (int i=; i<n; i++)
{
int x=ra(),y=ra();
insert(x,y); insert(y,x);
}
for (int x=; x<=n; x++)
{
memset(s1,,sizeof(s1));
memset(s2,,sizeof(s2));
for (int i=head[x];i;i=e[i].next)
{
dfs(e[i].to,x,);
for (int j=; j<=mx; j++)
{
ans+=s2[j]*tot[j];
s2[j]+=s1[j]*tot[j];
s1[j]+=tot[j];
}
for (int j=; j<=mx; j++) tot[j]=;
}
}
cout<<ans;
return ;
}