BZOJ 4011: [HNOI2015]落忆枫音

第一眼看是语文题？

读了一段感觉不对劲啊这不符合出题人的语文水平

百度了一下，WTF？？？？？？

不愧是老司机啊，省选都能开车，太厉害了

好了看题目，有向图生成树计数？

显然基尔霍夫矩阵在有向图上的应用我并不会

不过原图是个DAG

我们考虑一下朱-刘算法的第一步，给除根外每个点选个入边

显然DAG上会构成一棵有根树，于是答案就是除根节点外其他节点的入度的乘积啦

但是万恶的出题人显然不会让我们这么简单A掉

于是他加了一条边，那么图中就可能有环了

但是由于每个点只有一条入边，所以只能是基环外向树

那么问题就变成了统计基环外向树的个数

显然当环确定了之后给环外的每个点都找一条入边，就会形成基环外向树

设u为环上的点，v为树上的点，那么这个环的答案就是degree[v]之积也就是所有点入度的乘积再除去degree[u]之积

并且我们可以发现环上必然存在x->y这条边，并且在原图中有y->x的路径

那么我们令f[u]为原图中y到u点的路径总数。

把每个点度数的逆元作为点权，令g[u]为原图中y到u点的路径点权积之和

 Πdegree[u]*f[x]*g[x]即为基环外向树的个数

y为1的时候要特判一下，实现的时候可以把f和g搞到一起去

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=100000+5;
typedef long long ll;
const int p=(1e9)+7;
struct Edge{int to,next;}e[N<<1];
int head[N],cnt;
void ins(int u,int v){
	e[++cnt]=(Edge){v,head[u]};head[u]=cnt;
}
ll f[N],inv[N<<1];
int du[N],n,deg[N];
void topodp(){
	queue<int>q;
	for(int i=1;i<=n;i++)
	if(!deg[i])q.push(i);
	while(!q.empty()){
		int u=q.front();q.pop();
		f[u]=f[u]*inv[du[u]]%p;
		for(int i=head[u];i;i=e[i].next){
			int v=e[i].to;
			deg[v]--;
			f[v]=(f[v]+f[u])%p;
			if(!deg[v])q.push(v);
		}
	}
}
int main(){
	//freopen("a.in","r",stdin);
	int m,x,y;scanf("%d%d%d%d",&n,&m,&x,&y);
	inv[1]=1;
	for(int i=2;i<=m+1;i++)
	inv[i]=(-inv[p%i]*(p/i)%p+p)%p;
	deg[y]++;
	while(m--){
		int u,v;scanf("%d%d",&u,&v);
		ins(u,v);deg[v]++;
	}
	for(int i=1;i<=n;i++)du[i]=deg[i];
	ll ans=1;
	for(int i=2;i<=n;i++)ans=ans*deg[i]%p;
	deg[y]--;
	if(y==1){printf("%lld\n",ans);return 0;}
	f[y]=ans;
	topodp();
	ans=(ans-f[x]+p)%p;
	printf("%lld\n",ans);
	return 0;
}