【DP】BZOJ4011[HNOI2015]落忆枫音

【前言】
可能有人会问，4010去哪了？太水了被喝掉了。

【题目】
BZOJ
给定一幅有向无环图，现在加入一条边$(x,y)$，求加入后以$1$为根的树形图个数。
$n\le 10^5,m\le 2\times 10^5$

【解题思路】
如果不考虑新加入这条边，那么除根节点外每个点选一条入边，由于没有环，一点形成一个树形图，因此答案为${\prod }_{i=2}^{n}i{n}_i$
加入这条边后，我们仍用这个计算方式，但是要将不合法方案减去，即减去成环的情况。考虑其贡献，即环上，每个点选择环上对应的父亲，其他点随便选择父亲，即$\prod _{i,i\notin Circle}i{n}_i$
现在考虑计算每个点的贡献，设$f_i$表示从$y$到$i$所有路径产生的贡献，即
$$\sum _{S:Pathy\to i}\prod _{j,j\notin S}i{n}_j$$
那么可以有转移：
$f_i=\frac{{\sum }_{j,(j,i)\in E}{f}_j}{i{n}_i}$
初始值$f_y=\frac{{\prod }_{i=2}^{n}i{n}_i}{i{n}_y}$，按拓扑序进行转移即可。
$y=1$时要特判。

【参考代码】

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int N=2e5+10,mod=1e9+7;
int n,m,s,t,tot;
int head[N],in[N],du[N];
ll ans,f[N],inv[N];
queue<int>q;

int read()
{
	int ret=0;char c=getchar();
	while(!isdigit(c)) c=getchar();
	while(isdigit(c)) ret=ret*10+(c^48),c=getchar();
	return ret;
}

struct Tway{int v,nex;}e[N];
void add(int u,int v)
{
	e[++tot]=(Tway){v,head[u]};head[u]=tot;
	in[v]++;du[v]++;
}

void topsort()
{
	f[t]=ans;
	for(int i=1;i<=n;++i) if(!in[i]) q.push(i);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		f[x]=f[x]*inv[du[x]]%mod;
		for(int i=head[x];i;i=e[i].nex)
		{
			int v=e[i].v;
			f[v]=(f[v]+f[x])%mod;--in[v];
			if(!in[v]) q.push(v);
		}
	}
	printf("%lld\n",(ans-f[s]+mod)%mod);
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("BZOJ4011.in","r",stdin);
	freopen("BZOJ4011.out","w",stdout);
#endif
	inv[0]=inv[1]=1;for(int i=2;i<N;++i) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
	n=read();m=read();s=read();t=read();
	for(int i=1,x,y;i<=m;++i) x=read(),y=read(),add(x,y);
	++du[t];ans=1;
	for(int i=2;i<=n;++i) ans=ans*du[i]%mod;
	if(t==1) printf("%lld\n",ans);
	else topsort();
	return 0;
}