HDU4705 Y

描述

样例输入

4
1 2
1 3
1 4

样例输出

1

题意：
给一棵树问有多少三个为一组的点不在一条线上。

思路：

求三点在同一线上貌似更好求，对于一个点，假设它有2个子节点（左右），则以它为中心的相连三点个数为左子节点树上的所有节点个数乘上所有节点个数减去该点已经累加的节点个数，右子节点同样操作最后答案就是所有3个一组的组合数减去三个点在同一线上的组数。

#include<stdio.h>
#include<string.h>
#pragma comment(linker, "/STACK:16777216")
const int Ms=1e5+5;
struct node
{
    int pt,nex;
}E[Ms*2];
typedef long long ll;
int head[Ms],n,cnt;
int dp[Ms];
ll ans;
void dfs(int u,int fa)
{
    int v;
    dp[u]=1;
    for(int i=head[u];i!=-1;i=E[i].nex)
    {
        v=E[i].pt;
        if(v==fa)continue;
        dfs(v,u);
        dp[u]+=dp[v];
        ///printf("%d %d\n",n-dp[u],dp[v]);
        ans+=1LL*(n-dp[u])*dp[v];
    }
}
void init()
{
    ans=cnt=0;
    memset(head,-1,sizeof head);
}
void add_edge(int u,int v)
{
    E[cnt]=(node){v,head[u]};
    head[u]=cnt++;
}
int main()
{
    while(~scanf("%d",&n))
    {
        int u,v;
        init();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            add_edge(u,v);
            add_edge(v,u);
        }
        dfs(1,-1);
        ll tmp=1LL*n*(n-1)*(n-2)/6;
        printf("%lld\n",tmp-ans);
    }
    return 0;
}

若有什么错误，欢迎指正^ _ ^ 。