HDU--4705

题目：

Y

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=4705

分析：树形dp的思想，枚举中间点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<stack>
using namespace std;
#define maxn 100005
#define LL __int64
#pragma comment(linker, "/STACK:16777216")
vector<int>e[maxn];
LL ans,sum,n;
bool vis[maxn];
int dfs(int s)
{
    vis[s]=true;
    int son,t=0;
    for(int i=0;i<e[s].size();i++)
    {
        int to=e[s][i];
        if(!vis[to])
        {
            son=dfs(to);
            t+=son;
            ans+=(LL)(n-t-1)*son;
        }
    }
    return t+1;
}
int main()
{
    while(scanf("%I64d",&n)!=EOF)
    {
        int u,v,m;m=int(n);
        for(int i=0;i<maxn;i++)e[i].clear();
        for(int i=0;i<m-1;i++)
        {
            scanf("%d%d",&u,&v);
            e[u].push_back(v);
            e[v].push_back(u);
        }
        memset(vis,false,sizeof(vis));
        ans=0;
        LL p=dfs(1);
        sum=n*(n-1)*(n-2)/6;
        printf("%I64d
",sum-ans);
    }
    return 0;
}