洛谷3244 BZOJ4011 HNOI2015 落忆枫音 图论 dp

题目链接

题意：
给你一个n个点m条边的有向无环图，加了一条从x到y的边，求以1为根的有向生成树个数，对1e9+7取模。n<=10w,m<=20w

题解：
首先我们先不管加进来的那条边，先考虑对于一个DAG如何求有向生成树个数。我们会发现，答案是除了根的每个点的入度的乘积，原因是在生成树上每个点有一个入度就有一种父节点的选择方案，点与点之间满足乘法原理。

那么接下来我们考虑加进来的这条边对答案的影响。我们发现，加进来一条边之后可能会形成环，我们原来计算生成树个数的时候每个点有几个入度就可以乘多少是因为我们保证了图的连通并且无环，所以这里我们要用加进来这条边后的总方案数减去其中形成环的方案数。一个环出现的次数是所有点去掉环上的点之后，其余点的入度的乘积。由于新加入一条边之后形成的环可能不只一个，并且形成的环一定会包含新加进来的边，所以我们用建形出原图的反图的方法，从y到x进行图上dp。我们考虑一个环我们可以先预处理出总方案数，然后除以环上的每个点的度数来算出一个环要减去的答案（模数是1e9+7可以算逆元），环与环之间要减的答案是满足加法原理的，我们可以列出dp式子$f[x]=\frac{{\sum }_{x->y}f[y]}{degree[x]}$，其中在点$y$处设置初始值为原答案除以$y$的度数。于是我们就可以用一种记忆化搜索的办法求出答案，最后再用总方案减去形成环的方案即可。

代码：

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

int n,m,xx,yy,hed[100010],cnt,vis[100010];
long long ans,du[100010],s,f[100010];
const long long mod=1e9+7;
struct node
{
 int to,next;
}a[200010];
inline int read()
{
 int x=0;
 char s=getchar();
 while(s>'9'||s<'0')
 s=getchar();
 while(s>='0'&&s<='9')
 {
  x=x*10+s-'0';
  s=getchar();
 }
 return x;
}
inline void add(int from,int to)
{
 a[++cnt].to=to;
 a[cnt].next=hed[from];
 hed[from]=cnt;
}
inline long long ksm(long long x,long long y)
{
 long long res=1;
 while(y)
 {
  if(y&1)
  res=res*x%mod;
  x=x*x%mod;
  y>>=1;
 }
 return res;
}
inline void dfs(int x)
{
 vis[x]=1;
 if(x==yy)
 {
  f[x]=s*ksm(du[x],mod-2)%mod;
  return;
 } 
 for(int i=hed[x];i;i=a[i].next)
 {
  int y=a[i].to;
  if(!vis[y])
  dfs(y);
  f[x]=f[x]+f[y];
  if(f[x]>=mod)
  f[x]-=mod;
 }
 f[x]=f[x]*ksm(du[x],mod-2)%mod;
}
int main()
{
 scanf("%d%d%d%d",&n,&m,&xx,&yy);
 for(int i=1;i<=m;++i)
 {
  int x=read(),y=read();
  add(y,x);
  du[y]++;
 }
 s=1;
 ans=1;
 ++du[1];
 for(int i=1;i<=n;++i)
 {
  s=s*du[i]%mod;
  if(i==yy)
  ans=ans*(du[i]+1)%mod;
  else
  ans=ans*du[i]%mod;
 } 
 dfs(xx);
 ans=(ans-f[xx]+mod)%mod;
 printf("%lld\n",ans);
 return 0;
}