bzoj 4011: [HNOI2015]落忆枫音

题意 ：

问一个dag+一条边的图的生成树个数，树边为父亲连向儿子，1为根。

题解：

dag的生成树计数显然就是${\Pi }_{i=2}^{n}deg(i)$。
当我们多了一条边，再套用上式的话，就会算出不合法的情况。
这些情况显然就是含有包含额外边的环。
所以相当于与求出$Y->X$的路径，除去它们的贡献，再求个和。
dp即可。
code：

#include<cstdio>
#include<queue>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
const LL mod=1e9+7;
queue<int> q;
struct node{
    LL y,next;
}a[200010];LL len=0,last[100010];
LL n,m,X,Y;
void ins(LL x,LL y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
LL pow(LL a,LL b)
{
    LL ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;b>>=1;
    }
    return ans;
}
LL deg[100010],d[100010],ans=1,f[100010];
bool vis[100010];
void dfs(LL x)
{
    if(vis[x]) return;
    vis[x]=true;
    for(LL i=last[x];i;i=a[i].next)
    {
        LL y=a[i].y;d[y]++;
        dfs(y);
    }
}
int main()
{
    scanf("%lld %lld %lld %lld",&n,&m,&X,&Y);swap(X,Y);
    for(LL i=1;i<=m;i++)
    {
        LL x,y;scanf("%lld %lld",&x,&y);
        ins(x,y);deg[y]++;
    }
    deg[X]++;
    for(LL i=2;i<=n;i++) (ans*=deg[i])%=mod;
    memset(vis,false,sizeof(vis));
    dfs(X);q.push(X);f[X]=pow(deg[X],mod-2);
    while(!q.empty())
    {
        LL x=q.front();q.pop();
        for(LL i=last[x];i;i=a[i].next)
        {
            LL y=a[i].y;
            d[y]--;if(d[y]==0) q.push(y);
            (f[y]+=f[x]*pow(deg[y],mod-2)%mod)%=mod;
        }
    }
    if(X!=1) (ans-=(ans*f[Y]))%=mod;
    printf("%lld",(ans+mod)%mod);
}