hdu 4705 树形dp

/*
题目大意：给一棵n个节点的树，求任意三个点两两无直系亲属关系的种数。
可以求反方向的，对于每个节点i使他选,他的一个子树中选一个，再在剩余的任意选一个。
*/
#pragma comment(linker, "/STACK:16777216")
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int maxn=100005;
typedef __int64 LL;
int num[maxn],n;LL ans;
vector<int>f[maxn];

void dfs(int u,int pre)
{
    num[u]=1;
    for(int i=0;i<f[u].size();i++)
    {
        int v=f[u][i];
        if(v==pre) continue;
        dfs(v,u);
        num[u]+=num[v];
        ans+=(LL)num[v]*(n-num[u]);    
    }
    return ;
}
int main()
{
    int i,a,b;
    while(~scanf("%d",&n))
    {
        for(i=1;i<=n;i++) f[i].clear();
        for(i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            f[a].push_back(b);
            f[b].push_back(a);
        }
        memset(num,0,sizeof(num));
        ans=0;
        dfs(1,-1);
        printf("%I64d
",(LL)n*(n-1)*(n-2)/6-ans);
    }
    return 0;
}