2013 多校第九场 hdu 4705 Y（枚举 + 树上的计数）

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

题目大意：给你一棵树，让你求出不在同一条简单链的三个点的个数。

思路：枚举这三个点的中点，我们默认让1号节点做树根，设以某个节点u的子节点为根节点的子树大小为 v1、v2、v3。。，则另外的点有 （n - size[ u ]），然后把依次过来把这些点乘一下就好了。

这道题目我竟然写了2个小时，中间爆 int 还错了几次。。。 不想说什么了，挫啊。。。 = =

代码如下：

#pragma comment(linker, "/STACK:16777216")
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef __int64 lld;

const int MAXN = 100011 ;

int n;

struct Edge
{
    int t,next;
} edge[MAXN<<1];

int tot,head[MAXN];

void add_edge(int s,int t)
{
    edge[tot].t = t;
    edge[tot].next = head[s];
    head[s] = tot++;
}

int size[MAXN];

lld ans;

void dfs(int u,int fa)
{
    size[u] = 1;
    for(int e = head[u];e!=-1;e = edge[e].next)
    {
        int v = edge[e].t;
        if(v==fa) continue;
        dfs(v,u);
        ans += (lld)(size[u] - 1)*size[v];
        size[u] += size[v];
    }
    ans += (lld)(size[u] - 1)*(n - size[u]);
}

int main()
{
    while(~scanf("%d",&n))
    {
        tot=0;
        memset(head,-1,sizeof(head));
        int a,b;
        for(int i= 1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            add_edge(a,b);
            add_edge(b,a);
        }
        ans = 0;
        dfs(1,0);
        ans = (lld)n*(n - 1)*(n - 2)/(lld)6 - ans;
        printf("%I64d\n",ans);
    }
    return 0;
}

/*
5
1 2
1 3
2 4
2 5

6
1 2
1 3
2 4
2 5
3 6

7
1 2
1 3
2 4
2 5
3 6
3 7

*/