题目求的是非simple path 的个数,那么倒过来想,总数-simple path 数 = 我们所求的。 总数为C(n,3);
simple path 数,枚举每个点作为根时,每个子树的结点通过根的 都是 simple path,求和一下即可。
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
#define clr(a, x) memset(a, x, sizeof(a))
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define REP(i,a,b) for(int i=a;i<=b;i++)
typedef __int64 lld;
const int maxn = 100010;
//vector<int>g[maxn];
struct edge
{
int v,next;
}g[maxn<<1];
int head[maxn],e;
void add(int u,int v)
{
g[e].v=v;
g[e].next=head[u];
head[u]=e++;
}
lld ans;
int dp[maxn];
int n;
void dfs(int u,int p)
{
dp[u]=1;
int sum=n-1;
//int sz=g[u].size();
for(int i=head[u];i!=-1;i=g[i].next)
{
int v=g[i].v;
if(v==p) continue;
dfs(v,u);
dp[u]+=dp[v];
ans+=(lld)(sum-dp[v])*dp[v];
sum-=dp[v];
}
}
int main()
{
//int n;
while(~scanf("%d",&n))
{
memset(head,-1,sizeof(head));
e=0;// for(int i=0;i<=n;i++) g[i].clear();
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
add(u,v);//g[u].push_back(v);
add(v,u);//g[v].push_back(u);
}
ans=0;
dfs(1,0);
ans=(lld)(n)*(n-1)*(n-2)/6-ans;
printf("%I64d\n",ans);
}
}