2018 icpc 徐州现场赛G-树上差分+组合数学-大佬的代码

现场赛大佬打印的代码，观摩了一哈。

写了注释，贴一下，好好学习。%%%PKU

代码：

//树上差分(LCA)
#include<bits/stdc++.h>

#define For(i,x,y) for (int i=x;i<y;i++)
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define lf else if

#define dprintf(...) fprintf(stderr,__VA_ARGS__)
using namespace std;

typedef long long ll;
typedef double db;
typedef pair<int,int> pii;
typedef vector<int> Vi;

int IN(){//读入挂
    int c,f,x;
    while (!isdigit(c=getchar())&&c!='-');c=='-'?(f=1,x=0):(f=0,x=c-'0');
    while (isdigit(c=getchar())) x=(x<<1)+(x<<3)+c-'0';return !f?x:-x;
}

const int p=1e9+7;
const int N=3e5+19;

int Pow(int a,int b){//快速幂
    int res=1;
    for (;b;b>>=1,a=1ll*a*a%p) if (b&1) res=1ll*res*a%p;
    return res;
}

struct Edge{
    int y,nxt;
} E[N*2];
int fac[N],inv[N];
int las[N],fa[20][N],A[N],B[N],dep[N];
int n,m,cnt,x,y,z,ans,k;

int C(int n,int m){//组合数
    if (n<m) return 0;
    return 1ll*fac[n]*inv[m]%p*inv[n-m]%p;
}

void Link(int x,int y){//链式前向星存图
    E[cnt]=(Edge){y,las[x]};las[x]=cnt++;
    E[cnt]=(Edge){x,las[y]};las[y]=cnt++;
}

void dfs(int x){//LCA的dfs
    for (int i=las[x],y;~i;i=E[i].nxt)
        if ((y=E[i].y)!=fa[0][x]){
            fa[0][y]=x;
            dep[y]=dep[x]+1;
            dfs(y);
        }
}

int LCA(int x,int y){//LCA(ST)
    if (dep[x]>dep[y]) swap(x,y);
    for (int i=dep[y]-dep[x],k=0;i;i>>=1,k++) if(i&1) y=fa[k][y];
    if(x==y) return x;
    for (int i=19;~i;i--) if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
    return fa[0][x];
}

void Dfs(int x){//树上差分的dfs，从根节点深搜，回溯时将其本身的权值加上所有子节点的权值
    for (int i=las[x],y;~i;i=E[i].nxt)
        if ((y=E[i].y)!=fa[0][x]){//筛掉父节点
            Dfs(y);
            A[x]+=A[y];//累加权值和
            B[x]+=B[y];
        }
}

void Main(){
    n=IN(),m=IN(),k=IN();
    For(i,1,n+1) las[i]=-1,A[i]=B[i]=0;
    cnt=0;
    For(i,1,n) Link(IN(),IN());
    dfs(1);
    For(i,1,20) For(x,1,n+1) fa[i][x]=fa[i-1][fa[i-1][x]];
    For(i,1,m+1){
        x=IN(),y=IN();
        z=LCA(x,y);
        A[x]++,A[y]++,A[z]--,A[fa[0][z]]--;
        B[x]++,B[y]++,B[z]-=2;//起点终点权值+1，lca权值-2
    }
    Dfs(1);
    ans=0;
//    cout<<"--------"<<endl;
//    for(int i=1;i<=n;i++)
//        cout<<i<<" "<<A[i]<<endl;
//    cout<<"--------"<<endl;
//    for(int i=2;i<=n;i++)
//        cout<<i<<" "<<B[i]<<endl;
//    cout<<"--------"<<endl;
    For(i,1,n+1){
        ans=(ans+C(A[i],k))%p;
    }
    For(i,2,n+1){
        ans=(ans-C(B[i],k)+p)%p;
    }
    printf("%d
",ans);
}

int main(){
    fac[0]=1;
    For(i,1,N) fac[i]=1ll*fac[i-1]*i%p;
    inv[N-1]=Pow(fac[N-1],p-2);
    for(int i=N-1;i;i--) inv[i-1]=1ll*inv[i]*i%p;
    for(int T=IN();T--;) Main();
}

/*
1
3 6 2
1 2
1 3
1 1
2 2
3 3
1 2
1 3
2 3
*/

OK.