BZOJ 4011 HNOI2015 落忆枫音

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=4011

题目很长，写得也很有诗意与浪漫色彩，让我们不禁感叹出题人是一个多么英俊潇洒的人。

所以题目大意就是：

　　给一个有向无环图，问加上一条我给定的边后，有多少个以1为根的树形图？n<=1e5,m<=2*1e5

原图无重边，加上的边可以为自环。

　　

首先先来解决一个问题：

　　一个有向无环图的树形图有多少个呢？

相当于除了根节点以外，每个点随意取一个入度，为什么就一棵树呢？每个点选一个父亲，并且保证没有环，所以是一棵树。

好的，对于有向无环图就一定是这样的，那么若加入一条边<x,y>，

那么就可能再选择的过程中选出一个环[就是祖先的父亲是自己这种情况]。那么我们就需要删去这种情况。

首先要出现环，则必定包括了<x,y>这条边，剩下的部分是原图中y->x的一条路径，要求这条路径上的点必须选择一条路径使得构成一个环。而其它的点可以随意选。

设F(i)表示y->i上的点所成路径必须选择一条能构成环路径的方案数。

有初始值：

递推式：

　　

相当于j可以选择延续所有从y走来的i的路径，但是要除以j原本可以选择的路径总数。

最后答案=所有方案[入度乘积] - F(x)。

听说ZZD去年就想出来了这题，不过因为不会打逆元而失之交臂[不然就是省队了？%%%]

为什么会有逆元？...额我好像忘记说方案数要取模了...

那么这题中为了加速运算，就用了一个O(n)递推求1...n的所有逆元的方法，我也算是普及一下吧？...

画图的时候没太注意大小写，不要太在意...

那么i的逆元可以由p mod i的逆元得知了，这显然是一个已知的数，所以每次递推都是O(1)的，边界：1的逆元=1。

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

inline int in(){
    int x=;char ch=getchar();
    while(ch>'' || ch<'') ch=getchar();
    while(ch>='' && ch<='') x=x*+ch-'',ch=getchar();
    return x;
}

const int maxn=;
const int mod=1e9+;
typedef long long ll;

struct Node{
    int data,next;
}node[maxn<<];

#define now node[point].data
#define then node[point].next

int n,m,ans,cnt,x,y;
int head[maxn],d[maxn],ld[maxn];
int f[maxn],stack[maxn],top;
int inv[maxn<<];

void init(){
    inv[]=;
    for(int i=;i<=m;i++)
        inv[i]=(mod-(ll)inv[mod%i]*(mod/i)%mod)%mod;
}

void add(int u,int v){
    node[cnt].data=v;node[cnt].next=head[u];head[u]=cnt++;d[v]++;
}

int power(int a,int k){
    int ans=;
    for(;k;k>>=){
        if(k&) ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
    }
    return ans;
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("maple.in","r",stdin);
    freopen("maple.out","w",stdout);
#endif

    int u,v;

    n=in(),m=in();
    x=in(),y=in();
    init();
    for(int i=;i<=n;i++) head[i]=-;
    for(int i=;i<=m;i++)
        u=in(),v=in(),add(u,v);
    d[y]++;
    ans=;
    for(int i=;i<=n;i++) ans=(ll)ans*d[i]%mod;
    if(y==){ printf("%d",ans);return ;}
    else f[y]=(ll)ans*inv[d[y]]%mod;

    d[y]--;
    memcpy(ld,d,sizeof(d));
    for(int i=;i<=n;i++) if(!d[i]) stack[++top]=i;

    while(top){
        u=stack[top--];
        for(int point=head[u];point!=-;point=then){
            if(--d[now]==)
                stack[++top]=now;
            f[now]=(f[now]+(ll)f[u]*inv[ld[now]]%mod)%mod;
        }
    }

    ans=(ans-f[x]+mod)%mod;
    printf("%d",ans);
    return ;
}