hdu 4705 Y(树形DP)

题目链接：hdu 4705 Y

题意：

给你一棵n个节点的树，问你不在同一条路径上的三点对的对数。

题解：

n个节点任选三个点C_n³ ,然后我们可以dp出在同一条路径上的三点对，然后减一减

#include<cstdio>
#pragma comment(linker, "/STACK:16777216")
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;

const int N=1e5+7;
int n,g[N],v[2*N],nxt[N*2],ed,x,y,sz[N];
ll ans;

void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}

void dfs(int x,int fa)
{
    sz[x]=1;
    for(int i=g[x];i;i=nxt[i])
        if(v[i]!=fa)dfs(v[i],x),sz[x]+=sz[v[i]];
}

void dfs2(int x,int fa)
{
    for(int i=g[x];i;i=nxt[i])
        if(v[i]!=fa)ans+=1ll*sz[v[i]]*(sz[x]-1-sz[v[i]]);
}

int main()
{
    while(~scanf("%d",&n))
    {
        ed=0;
        F(i,1,n)g[i]=0;
        F(i,1,n-1)scanf("%d%d",&x,&y),adg(x,y),adg(y,x);
        ans=0,dfs(1,0),dfs2(1,0);
        printf("%lld
",1ll*n*(n-1)*(n-2)/6-ans/2);
    }
    return 0;
}