HDOJ 4705 Y

看了一下别人的做法，是求出3点构成链的方案数 ，然后总方案数减去它。

对于一个父节点u,通过它可以产生的3点成链有这样几种，一种是3点以u和u以下某分支的子节点构成，还有种情况是以u为中点，挑两个分支的子节点，最后一种情况是在一个分支上挑2个点，另一分支挑1个点。

#pragma comment(linker, "/STACK:16777216")
#include <cmath>
#include <queue>
#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
vector<int> eg[100005];
__int64 son[100005], com[100005];
__int64 ans, k;
void dfs(int pre,int cur)
{
    int i, j;
    __int64 cnt=0, cnt2=0;
    son[cur]=1;
    for(i=0;i<eg[cur].size();i++){
        if(eg[cur][i]==pre) continue;
        dfs(cur,eg[cur][i]);
        son[cur]+=son[eg[cur][i]];
        com[cur]+=com[eg[cur][i]]+son[eg[cur][i]];
        ans+=cnt*son[eg[cur][i]];
        ans+=com[eg[cur][i]];
        ans+=cnt*com[eg[cur][i]];
        ans+=cnt2*son[eg[cur][i]];
        cnt2+=com[eg[cur][i]];
        cnt+=son[eg[cur][i]];
    }
}
int main() {
    int i, a, b, n;
    while(~scanf("%d",&n)){
        for(i=1;i<=n;i++) eg[i].clear(), son[i]=com[i]=0;
        for(i=1;i<n;i++){
            scanf("%d%d",&a,&b);
            eg[a].push_back(b);
            eg[b].push_back(a);
        }
        ans=0;
        k=n;
        dfs(-1,1);
        printf("%I64d\n",k*(k-1)*(k-2)/6-ans);
    }
    return 0;
}