【汕头市选2014】分叉 jzoj 3630 dfs

题目

给出一棵N 个点的树，点的编号是1， 2，。。。，N。

对于3 个点{a,b,c}，如果不存在一条简单路径同时经过a,b,c，那么{a,b,c}是一个分叉。

统计不同分叉的数量。

分析

比赛时爆int.

开了long long就对了

我们可以很方便的找到有一条简单路径经过{x,y,z}三元组的个数

然后拿所有的三元组个数去减就好了.

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#define maxn 200005
#define maxm 300005
#define INF 0x7fffffff;

using namespace std;

struct arr{
int x,y;
int next;
}edge[maxm];
int ls[maxn];
int edge_m;

long long son[maxn];
long long f[maxn];

long long n;

void add(int x,int y)
{
edge[++edge_m]=(arr){x,y,ls[x]},ls[x]=edge_m;
edge[++edge_m]=(arr){y,x,ls[y]},ls[y]=edge_m;
}

int dfs(int x,int r)
{
son[x]=1;
for (int i=ls[x];i;i=edge[i].next)
{
if (edge[i].y==r) continue;
dfs(edge[i].y,x);
son[x]+=son[edge[i].y];
}
int num=1;
for (int i=ls[x];i;i=edge[i].next)
{
if (edge[i].y==r) continue;
num+=son[edge[i].y];
f[x]+=son[edge[i].y]*(n-num);
}
}

int main()
{
freopen("fork.in","r",stdin);
freopen("fork.out","w",stdout);
scanf("%d",&n);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);
}
dfs(1,0);
long long ans=(long long)(n*(n-1)*(n-2)/6);
for (int i=1;i<=n;i++)
{
ans-=f[i];
}
cout<<ans;
}