CodeForces 288 D.Polo the Penguin and Trees（组合数学）

Description

给出一棵$n$个节点的树，问四元组$(a,b,c,d)$个数使其满足：

1.$1\le a<b\le n,1\le c<d\le n$

2.树上从$a$到$b$的简单路径与$c$到$d$的简单路径不经过相同节点

Input

第一行一整数$n$表示点数，之后$n-1$行每行两个整数$u,v$表示一条树边$(1\le n\le 80000)$

Output

输出满足条件的四元组个数

Sample Input

4
1 2
2 3
3 4

Sample Output

2

Solution

满足第一个条件的四元组数量为$C_n^2\cdot C_n^2$，考虑求不满足第二个条件的四元组数量，即两条路径有公共点，考虑每个点$u$作为两条路径交点中深度最小的公共点对不合法答案的贡献，记$size[u]$为以$u$为根的子树个数，$num_1$为经过点$u$且一端在$u$的子树中，另一端不在$u$子树中的路径数量，$num_2$表示经过点$u$且两端都在$u$的子树中的路径数量，这两个值都比较好求

$num_2=C_{size[u]}^2-\sum\limits_{fa[v]=u}C_{size[v]}^2$

$num_1=C_n^2-\sum\limits_{fa[v]=u}C_{size_v}^2-C_{size[fa[u]]}^2-num_2$

由于不可能存在两条路径使得这两条路径都是一端在$u$子树中一端不在，因为这样的话这两条路径公共点中深度最小的不会是$u$，至少也是$fa[u]$，故这两条路径必然是两种情况：一条路径一段在$u$子树一段不在另一条路径两端都在，或者是两条路径都是两端都在$u$子树中，故方案数为$2\cdot num_1\cdot num_2+num_2\cdot num_2$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=80005;
int n,size[maxn],fa[maxn];
vector<int>g[maxn]; 
void dfs(int u,int f)
{
    size[u]=1;
    fa[u]=f;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==f)continue;
        dfs(v,u);
        size[u]+=size[v];
    }
}
ll C(int n)
{
    return (ll)n*(n-1)/2;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v),g[v].push_back(u);
    }
    dfs(1,0);
    ll ans=C(n)*C(n);

    for(int u=1;u<=n;u++)
    {
        ll num1=C(n),num2=C(size[u]);
        for(int i=0;i<g[u].size();i++)
        {
            int v=g[u][i];
            if(v!=fa[u])num1-=C(size[v]),num2-=C(size[v]);
            else num1-=C(n-size[u]);
        }
        ans-=(2ll*num1*num2-num2*num2);
    } 
    printf("%I64d\n",ans);
    return 0;
}