BZOJ4011 HNOI2015落憶楓音(動態規劃+拓撲排序)
阿新 • • 發佈:2018-11-09
DAG中每個點選一條入邊就可以構成一棵有向樹,所以如果沒有環答案就是∏degreei。
考慮去掉含環的答案。可以看做把環縮點,剩下的點仍然可以任意選入邊。於是去除的方案數即為∏degreei/∏degreek,k為環上點。
環相當於考慮新加入邊的終點到起點的所有路徑。設f[i]為i為起點的所有路徑提供的上述貢獻,則f[i]=Σf[k]/degree[i]。拓撲排序之後dp即可。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; int read() { int x=0,f=1;char c=getchar(); while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();} while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } #define N 100010 #define M 200010 #defineP 1000000007 int n,m,p[N],u,v,degree[N],d[N],f[N],q[N],inv[N],t=0,ans=1; struct data{int to,nxt; }edge[M]; void addedge(int x,int y){t++;edge[t].to=y,edge[t].nxt=p[x],p[x]=t;} void topsort() { int head=0,tail=1;q[1]=1;memcpy(d,degree,sizeof(degree));d[v]--; while (tail<n) { int x=q[++head];for (int i=p[x];i;i=edge[i].nxt) { d[edge[i].to]--; if (!d[edge[i].to]) q[++tail]=edge[i].to; } } } int main() { #ifndef ONLINE_JUDGE freopen("bzoj4011.in","r",stdin); freopen("bzoj4011.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(),u=read(),v=read(); for (int i=1;i<=m;i++) { int x=read(),y=read(); addedge(x,y);degree[y]++; } inv[1]=1;for (int i=2;i<=n;i++) inv[i]=P-1ll*(P/i)*inv[P%i]%P; degree[v]++; for (int i=2;i<=n;i++) ans=1ll*ans*degree[i]%P; if (v==1) {cout<<ans;return 0;} topsort();f[u]=ans; for (int i=n;i>=1;i--) { for (int j=p[q[i]];j;j=edge[j].nxt) f[q[i]]=(f[q[i]]+f[edge[j].to])%P; f[q[i]]=1ll*f[q[i]]*inv[degree[q[i]]]%P; } cout<<(ans-f[v]+P)%P; return 0; }