2018徐州G.Rikka with Intersections of Paths

https://codeforces.com/group/NVaJtLaLjS/contest/255471/problem/G
题意：有一棵n个结点的树，现在给出m条树上的路径。现在要从这m条路径中选出k条路径，使得这k条路径至少有一个公共交点，问你总共有多少种方案数。
思路：
有一个重要的性质：一个树上任意两条路径如果有交点的话，那么这些交点中肯定有一个为两条路径中的一条路径的lca。
推论：一个树上任意k条路径如果有交点的话，那么这些交点中肯定有一个为k条路径中的至少一条路径的lca。
将此交点称为关键点。
那么我们枚举每一个关键点，设经过某点的路径数为s，其中有p条路径的LCA为这个点，那么此处的计数就是$\left(\genfrac{}{}{0ex}{}{s}{k}\right)-\left(\genfrac{}{}{0ex}{}{s-p}{k}\right)$。这样保证了每个组合只对答案贡献一次。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=300000+100;

int n,m,k;
vector<int> G[maxn];
int fa[maxn][20],deep[maxn],d[maxn];
long long ans;
int cnt[maxn],lca_cnt[maxn];
ll mod=1e9+7,fac[maxn];

void dfs(int u,int f)
{
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==f)continue;
        fa[v][0]=u;
        deep[v]=deep[u]+1;
        dfs(v,u);
    }
}

void lca_init()
{
    for(int i=1;i<=n;i++)for(int j=1;(1<<j)<n;j++)fa[i][j]=-1;
    for(int j=1;(1<<j)<n;j++)
        for(int i=1;i<=n;i++)
            if(fa[i][j-1]!=-1)
            {
                int a=fa[i][j-1];
                fa[i][j]=fa[a][j-1];
            }
}

int lca(int p,int q)
{
    int log;
    if(deep[p]<deep[q])swap(p,q);
    for(log=1;(1<<log)<=deep[p];log++);
    log--;
    int ans=-(1<<30);
    for(int i=log;i>=0;i--)
        if(deep[p]-(1<<i)>=deep[q])
        {
            p=fa[p][i];
        }
    if(p==q)return p; //lca为p或q
    for(int i=log;i>=0;i--)
        if(fa[p][i]!=-1 && fa[p][i]!=fa[q][i])
        {
            p=fa[p][i];
            q=fa[q][i];
        }
    return fa[p][0];//lca为fa[p]或fa[q]
}

int dfs2(int u,int f)
{
    cnt[u]=d[u];
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==f)continue;
        cnt[u]+=dfs2(v,u);
    }
    return cnt[u];
}

ll pow_mod(ll a,ll n)
{
    if(!n)return 1;
    ll x=pow_mod(a,n/2);
    x=x*x%mod;
    if(n&1)x=x*a%mod;
    return x;
}

int C(int n,int m)
{
    if(n<m)return 0;
    return fac[n]*pow_mod(fac[m],mod-2)%mod*pow_mod(fac[n-m],mod-2)%mod;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int T;
    cin>>T;
    fac[0]=1;
    for(int i=1;i<maxn;i++)fac[i]=(long long)fac[i-1]*i%mod;
    while(T--)
    {
        memset(cnt,0,sizeof(cnt));
        memset(lca_cnt,0,sizeof(lca_cnt));
        memset(d,0,sizeof(d));
        ans=0;
        cin>>n>>m>>k;
        for(int i=1;i<=n;i++)G[i].clear();
        int a,b;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        fa[1][0]=-1;
        deep[1]=0;
        dfs(1,-1);
        lca_init();
        while(m--)
        {
            scanf("%d%d",&a,&b);
            int x=lca(a,b);
            lca_cnt[x]++;
            d[a]++;d[b]++;
            d[x]--;
            if(fa[x][0]>0)d[fa[x][0]]--;
        }
        dfs2(1,-1);
        for(int i=1;i<=n;i++)
        {
            ans=(ans+C(cnt[i],k)-C(cnt[i]-lca_cnt[i],k)+mod)%mod;
        }
        cout<<ans<<endl;
    }
    return 0;
}