18 icpc徐州 G. Rikka with Intersections of Paths//lca+树上点差分+容斥 (组合数模板O(N)

31 篇文章 0 订阅
13 篇文章 0 订阅

https://codeforc.es/gym/102012/problem/G

题意:给一棵树,给树上m条边,求能选出k条使得k条至少有一个公共点的方案数。

思路:

https://blog.csdn.net/lee_w_j__/article/details/84780981

https://blog.csdn.net/NPU_SXY/article/details/84708278

wsl

 

/*   Author : Rshs
 *   Data : 2019-09-10-18.53
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const LL mod = 1e9+7;
const int MX = 3e5+5;
const int MXLOG = 22;
vector<int>g[MX];
int bei[MX][MXLOG],de[MX];
LL cf[MX],lc[MX];  //差分标记

void dfs(int v,int p){//所有都是向上走2^k步 所代表的含义
    for(auto i:g[v]) {
        if(i!=p){
            de[i]=de[v]+1;bei[i][0]=v;
            dfs(i,v);
        }
    }
}
void initLCA(int nn){
    bei[1][0]=-1;de[1]=0;dfs(1,-1);
    for(int k=0;k<MXLOG-1;k++){   //-1!!
        for(int v=1;v<=nn;v++){
            if(bei[v][k]<0)bei[v][k+1]=-1;
            else bei[v][k+1]=bei[bei[v][k]][k];
        }
    }
}
int LCA(int u,int v){
    if(de[u]>de[v]) swap(u,v);
    for(int k=0;k<MXLOG;k++){
        if((de[v]-de[u])>>k&1) v=bei[v][k];
    }
    if(u==v) return u;
    for(int k=MXLOG-1;k>=0;k--){
        if(bei[u][k]!=bei[v][k]){u=bei[u][k],v=bei[v][k];}
    }
    return bei[u][0];
}
void addcf(int u,int v,int c){//(x->y)的路径上的点权值加c
    int lca=LCA(u,v);
    cf[u]+=c,cf[v]+=c,cf[lca]-=c;
    if(bei[lca][0]!=-1) cf[bei[lca][0]]-=c;
}
void dfscf(int v,int p){  //标记转化成真实值后,cf[i]表示i到fa[i]的边的权值
    for(auto i:g[v]){
        if(i==p)continue;
        dfscf(i,v);
        cf[v]+=cf[i];
    }
}
LL infa[MX],inv[MX],fac[MX];
void CCinit(int n,LL p){
    fac[0]=1;
    infa[1]=inv[1]=fac[1]=1;
    for(LL i=2;i<=n;i++) fac[i]=fac[i-1]*i%p;
    for(LL i=2;i<=n;i++) inv[i]=(p-p/i)*inv[p%i]%p;
    for(LL i=2;i<=n;i++) infa[i]=infa[i-1]*inv[i]%p;
}
LL CC(int a,int b,LL p){ //c(b/a)
    if(a<0||b<0||a<b) return 0;
    if(b==0||a==b) return 1;
    return fac[a]*infa[b]%p*infa[a-b]%p;
}

void Main(int avg){
    int n,m,k;cin>>n>>m>>k;
    for(int i=1;i<=n;i++) g[i].clear(),cf[i]=lc[i]=0;
    for(int i=2;i<=n;i++){
        int sa,sb;scanf("%d%d",&sa,&sb);
        g[sa].push_back(sb),g[sb].push_back(sa);
    }
    initLCA(n);
    for(int i=1;i<=m;i++){
        int sa,sb;scanf("%d%d",&sa,&sb);
        addcf(sa,sb,1);lc[LCA(sa,sb)]++;
    }
    dfscf(1,-1);
    LL ans=0;
    for(int i=1;i<=n;i++){
        ans=(ans+CC(cf[i],k,mod)-CC(cf[i]-lc[i],k,mod))%mod;
    }
    cout<<(ans+mod)%mod<<'\n';
}
int main(){
    CCinit(3e5,mod);
    int cas;cin>>cas;for(int i=1;i<=cas;i++)Main(i);
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值