HNOI 2015 落忆枫音题解

最新推荐文章于 2022-09-16 17:58:48 发布

Hypoc_

最新推荐文章于 2022-09-16 17:58:48 发布

阅读量164

点赞数

分类专栏：题解_杂

本文链接：https://blog.csdn.net/a_forever_dream/article/details/108864600

版权

题解_杂专栏收录该内容

304 篇文章 6 订阅

订阅专栏

题目传送门

题目大意： 给出一个 $\text{DAG}$ ，然后往里面加一条额外边 $(s x, s y)$ ，询问有多少棵以 $1$ 为根的外向树。

题解

先考虑不选额外边的答案，令 $d_i$ 表示 $i$ 的入度，则答案为 $\prod_{i=2}^n d_i$ ，即每个点需要选择一条入边。

再考虑选额外边的答案：如果不考虑环，贡献就是 $\prod_{i=2,i\neq sy}d_i$ ，再减去其中成环的方案： $\text{DAG}$ 中一条 $s y$ 到 $s x$ 的路径，就对应着一个环，假设这个环中包含的点集为 $S$ ，那么这个环会被计算这么多次： $\prod_{i=2,i\not\in S}^n d_i$ 。

令 $f_i$ 表示从 $s y$ 出发到 $i$ 的所有路径的贡献和，考虑按拓扑序来转移，对于一条边 $(x, y)$ ， $f_x$ 对 $f_y$ 的贡献为 $\dfrac {f_x} {d_y}$ ，最后使答案减去 $f_{sx}$ 即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 100010
#define mod 1000000007

int n,m,sx,sy,ans=0;
struct edge{int y,next;}e[maxn<<1];
int first[maxn],len=0;
void buildroad(int x,int y){e[++len]=(edge){y,first[x]};first[x]=len;}
int d[maxn],invd[maxn];
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
int f[maxn],sta[maxn],t=0;
void add(int &x,int y){x=(x+y>=mod?x+y-mod:x+y);}
void dec(int &x,int y){x=(x-y<0?x-y+mod:x-y);}
void solve(){
	sta[++t]=1;f[sy]=1;
	for(int i=2;i<=n;i++)
	if(i!=sy)f[sy]=1ll*f[sy]*d[i]%mod;
	while(t){
		int x=sta[t--];
		for(int i=first[x];i;i=e[i].next){
			int y=e[i].y;add(f[y],1ll*f[x]*invd[y]%mod);
			d[y]--;if(!d[y])sta[++t]=y;
		}
	}
}

int main()
{
	scanf("%d %d %d %d",&n,&m,&sx,&sy);
	for(int i=1,x,y;i<=m;i++){
		scanf("%d %d",&x,&y);
		buildroad(x,y);d[y]++;
	}
	for(int i=2;i<=n;i++)invd[i]=ksm(d[i],mod-2);
	int tmp=1;for(int i=2;i<=n;i++)tmp=1ll*tmp*d[i]%mod;
	ans=tmp;
	if(sx!=sy&&sy!=1){
		tmp=1;for(int i=2;i<=n;i++)if(i!=sy)tmp=1ll*tmp*d[i]%mod;
		add(ans,tmp);
		solve();
		dec(ans,f[sx]);
	}
	printf("%d",ans);
}