【bzoj4011】【HNOI2015】落忆枫音【树形DP】

bzoj链接
小视野链接
###题目大意
给出一个DAG，现在新加一条边（给出：$x\to y$），求以1为根生成的外向树的个数。（题目保证图中从1节点能到达所有节点，可能有自环）
###题解
这种解法是大神zjr想出来的：。
首先算出每个点的入度$de{g}_i$
如果没有环，答案就为${\prod }_{i=2}^{n}de{g}_i$，因为每个点可以任意选择一条入边。
若加上的边仍没有环或不能选上（连上1或自环），答案也是如此。
然而懒得判有环无环，这两种情况合在一起考虑，这时答案为不加这条边的方案加上不加的方案。不加的方案数显然是${\prod }_{i=2}^{n}de{g}_i$（deg中不计新加的贡献）。
考虑加上这条边的答案。若无环：先单独把这条边取出来，把树分为以1为根的子树和以$y$为根的子树，子树中随便取，必须取$x\to y$，$y$的入边不能取，答案为${\prod }_{i=2,i̸=y}^{n}de{g}_i=\frac{{\prod }_{i=2}^{n}de{g}_i}{de{g}_y}$。若有环，规定只选一条1至$x$的路径，其他都不选，这样答案为$|S|\ast {\prod }_{i=2,i̸=y,i̸\in S}^{n}de{g}_i=\frac{1}{de{g}_y}\ast {\prod }_{i=2}^{n}de{g}_i\ast |S|\ast {\prod }_{i\in S}\frac{1}{de{g}_i}$(S为从根节点不经过y到达x的路径点集)。后一部分可以用DP算出来$dp[u]={\sum }_{u\to v}\frac{dp[v]}{deg[v]}$，边界为$dp[i]=(i==X)$。（注意不能进过y）。
由于在无环的情况下，这个DP值不影响答案，所以一起计算更方便
###代码

//#pragma GCC optimize(3)
#include<iostream>
#include<iomanip>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set>
#include<map>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#define ll long long
#define db double
#define inf 100009
#define infm 100009
#define INF (ll)1e15
#define mod 1000000007LL
#define pi acos(-1)
#define rd(n) {n=0;char ch;int f=0;do{ch=getchar();if(ch=='-'){f=1;}}while(ch<'0'||ch>'9');while('0'<=ch&&ch<='9'){n=(n<<1)+(n<<3)+ch-48;ch=getchar();}if(f)n=-n;}
using namespace std;
int n,m,X,Y;

ll inv[inf];
ll getinv(ll x){
	int tmpx=(ll)x;
/*	if (inv[tmpx]){
		return inv[tmpx];
	}*/
	ll ans=1LL,k=mod-2LL;
	while (k){
		if (k&1LL){
			ans=(ans*x)%mod;
		}
		x=(x*x)%mod;
		k/=2LL;
	}
//	inv[tmpx]=ans;
	return ans;
}

struct edge{
	int y,nxt;
	edge(){}
	edge(int yy,int nn){
		y=yy,nxt=nn;
	}
}e[inf*2];
int head[inf],ecnt;

void addedge(int x,int y){
	ecnt++;
	e[ecnt]=edge(y,head[x]);
	head[x]=ecnt;
	return;
}

ll deg[inf],dp[inf];
bool vis[inf];

void dfs(int u){
	vis[u]=1;
	if (u==X){
		dp[u]=1LL;
		return;
	}
	else if (u==Y){
		return;
	}
	for (int i=head[u];i;i=e[i].nxt){
		int v=e[i].y;
		if (!vis[v]){
			dfs(v);
		}
		dp[u]=(dp[u]+getinv(deg[v])*dp[v]%mod)%mod;
	}
	return;
}

int main(){
	rd(n) rd(m) rd(X) rd(Y)
	int x,y;
	for (int i=1;i<=m;i++){
		rd(x) rd(y)
		addedge(x,y);
		deg[y]++;
	}
	ll ans=1LL;
	for (int i=2;i<=n;i++){
		ans=(ans*deg[i])%mod;
	}
	if (X==Y || Y==1){
		printf("%lld\n",ans);
		return 0;
	}
	dfs(1);
	ans=(ans+ans*dp[1]%mod*getinv(deg[Y])%mod)%mod;
	printf("%lld\n",ans);
	return 0;
}