bzoj3456 城市規劃
阿新 • • 發佈:2017-08-13
ret 化簡 include bzoj tdi 解決 總數 所在 繼續
Description
求含有n個點有標號的無向聯通圖的個數(沒有重邊),n<=130000
Input
3
Output
4
正解:$分治FFT$/多項式求逆。
並沒有權限號,但是某$oj$裏有這道題。。
我們考慮遞推,$f[i]$表示$i$個點的聯通圖個數,那麽用總數減去不合法的數量。
考慮枚舉$1$號點所在的聯通塊的點數,那麽我們可以得到:
$f[n]=2^{\binom{n}{2}}-\sum_{i=1}^{n-1}f[i]*\binom{n-1}{i-1}*2^{\binom{n-i}{2}}$
那麽這個式子把組合數拆開以後就能直接上分治$FFT$了,復雜度$O(nlog^{2}n)$
分治$FFT$:
1 #include <bits/stdc++.h> 2 #define il inline 3 #define RG register 4 #define ll long long 5 #define rhl (1004535809) 6 #define N (1000010) 7 #define G (3) 8 9 using namespace std; 10 11 int fac[N],ifac[N],inv[N],cal[N],w[N],rev[N],f[N],a[N],b[N],n;12 13 il int qpow(RG int a,RG ll b){ 14 RG int ans=1; 15 while (b){ 16 if (b&1) ans=1LL*ans*a%rhl; 17 a=1LL*a*a%rhl,b>>=1; 18 } 19 return ans; 20 } 21 22 il void pre(){ 23 fac[0]=fac[1]=ifac[0]=ifac[1]=inv[1]=cal[0]=cal[1]=1; 24 for (RG int i=2;i<=n;++i){25 fac[i]=1LL*fac[i-1]*i%rhl; 26 inv[i]=1LL*(rhl-rhl/i)*inv[rhl%i]%rhl; 27 ifac[i]=1LL*ifac[i-1]*inv[i]%rhl; 28 cal[i]=qpow(2,1LL*i*(i-1)>>1); 29 } 30 for (RG int i=1,v=1;i<=(n<<1);++v,i<<=1) 31 w[v]=qpow(G,(rhl-1)/(i<<1)); 32 return; 33 } 34 35 il void NTT(int *a,RG int n,RG int f){ 36 for (RG int i=0;i<n;++i) if (i<rev[i]) swap(a[i],a[rev[i]]); 37 for (RG int i=1,v=1;i<n;++v,i<<=1){ 38 RG int gn=w[v],x,y; 39 for (RG int j=0;j<n;j+=i<<1){ 40 RG int g=1; 41 for (RG int k=0;k<i;++k,g=1LL*g*gn%rhl){ 42 x=a[j+k],y=1LL*g*a[j+k+i]%rhl; 43 a[j+k]=x+y; if (a[j+k]>=rhl) a[j+k]-=rhl; 44 a[j+k+i]=x-y; if (a[j+k+i]<0) a[j+k+i]+=rhl; 45 } 46 } 47 } 48 if (f==1) return; reverse(a+1,a+n); RG int inv=qpow(n,rhl-2); 49 for (RG int i=0;i<n;++i) a[i]=1LL*a[i]*inv%rhl; return; 50 } 51 52 il void solve(RG int l,RG int r){ 53 if (l==r){ 54 f[l]=(cal[l]-1LL*fac[l-1]*f[l]%rhl+rhl)%rhl; 55 return; 56 } 57 RG int mid=(l+r)>>1; solve(l,mid); 58 RG int m=max(mid-l+1,r-mid),len,lg=0; 59 for (len=1;len<=(m<<1);len<<=1) ++lg; 60 for (RG int i=0;i<len;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1)),a[i]=b[i]=0; 61 for (RG int i=l;i<=mid;++i) a[i-l]=1LL*f[i]*ifac[i-1]%rhl; 62 for (RG int i=1;i<=r-l;++i) b[i]=1LL*cal[i]*ifac[i]%rhl; 63 NTT(a,len,1),NTT(b,len,1); for (RG int i=0;i<len;++i) a[i]=1LL*a[i]*b[i]%rhl; 64 NTT(a,len,-1); for (RG int i=mid+1;i<=r;++i) f[i]=(f[i]+1LL*a[i-l])%rhl; 65 solve(mid+1,r); return; 66 } 67 68 int main(){ 69 #ifndef ONLINE_JUDGE 70 freopen("city.in","r",stdin); 71 freopen("city.out","w",stdout); 72 #endif 73 cin>>n; pre(); solve(1,n); 74 printf("%d\n",f[n]); 75 return 0; 76 }
我們還可以繼續化簡,得到:
$\sum_{i=1}^{n}f[i]*\binom{n-1}{i-1}*2^{\binom{n-i}{2}}=2^{\binom{n}{2}}$
$\sum_{i=1}^{n}f[i]*(i-1)!^{-1}*2^{\binom{n-i}{2}}*(n-i)!^{-1}=2^{\binom{n}{2}}*(n-1)!^{-1}$
註意到上式其實是兩個多項式卷積等於第三個多項式的形式,那麽我們可以用多項式求逆來解決這個問題,復雜度$O(nlogn)$。
多項式求逆:
1 #include <bits/stdc++.h> 2 #define il inline 3 #define RG register 4 #define ll long long 5 #define rhl (1004535809) 6 #define N (1000010) 7 #define G (3) 8 9 using namespace std; 10 11 int fac[N],ifac[N],inv[N],cal[N],w[N],rev[N],a[N],b[N],invb[N],tmp[N],n,m; 12 13 il int qpow(RG int a,RG ll b){ 14 RG int ans=1; 15 while (b){ 16 if (b&1) ans=1LL*ans*a%rhl; 17 a=1LL*a*a%rhl,b>>=1; 18 } 19 return ans; 20 } 21 22 il void pre(){ 23 fac[0]=fac[1]=ifac[0]=ifac[1]=inv[1]=cal[0]=cal[1]=1; 24 for (RG int i=2;i<=n;++i){ 25 fac[i]=1LL*fac[i-1]*i%rhl; 26 inv[i]=1LL*(rhl-rhl/i)*inv[rhl%i]%rhl; 27 ifac[i]=1LL*ifac[i-1]*inv[i]%rhl; 28 cal[i]=qpow(2,1LL*i*(i-1)>>1); 29 } 30 for (RG int i=1,v=1;i<=(n<<1);++v,i<<=1) 31 w[v]=qpow(G,(rhl-1)/(i<<1)); 32 return; 33 } 34 35 il void NTT(int *a,RG int n,RG int f){ 36 for (RG int i=0;i<n;++i) if (i<rev[i]) swap(a[i],a[rev[i]]); 37 for (RG int i=1,v=1;i<n;++v,i<<=1){ 38 RG int gn=w[v],x,y; 39 for (RG int j=0;j<n;j+=i<<1){ 40 RG int g=1; 41 for (RG int k=0;k<i;++k,g=1LL*g*gn%rhl){ 42 x=a[j+k],y=1LL*g*a[j+k+i]%rhl; 43 a[j+k]=x+y; if (a[j+k]>=rhl) a[j+k]-=rhl; 44 a[j+k+i]=x-y; if (a[j+k+i]<0) a[j+k+i]+=rhl; 45 } 46 } 47 } 48 if (f==1) return; reverse(a+1,a+n); RG int inv=qpow(n,rhl-2); 49 for (RG int i=0;i<n;++i) a[i]=1LL*a[i]*inv%rhl; return; 50 } 51 52 il void getinv(int *a,int *b,RG int n){ 53 if (n==1){ b[0]=qpow(a[0],rhl-2); return; } getinv(a,b,n>>1); 54 for (RG int i=0;i<n;++i) tmp[i]=a[i],tmp[n+i]=0; 55 RG int lg=0,m=0; for (m=1;m<=n;m<<=1) ++lg; 56 for (RG int i=0;i<m;++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1)); 57 NTT(tmp,m,1),NTT(b,m,1); 58 for (RG int i=0;i<m;++i) tmp[i]=((b[i]<<1)-1LL*b[i]*b[i]%rhl*tmp[i]%rhl+rhl)%rhl; 59 NTT(tmp,m,-1); for (RG int i=0;i<n;++i) b[i]=tmp[i],b[n+i]=0; return; 60 } 61 62 int main(){ 63 #ifndef ONLINE_JUDGE 64 freopen("city.in","r",stdin); 65 freopen("city.out","w",stdout); 66 #endif 67 cin>>n; pre(); for (m=1;m<=n;m<<=1); 68 for (RG int i=1;i<=n;++i) a[i]=1LL*cal[i]*ifac[i-1]%rhl; 69 for (RG int i=0;i<=n;++i) b[i]=1LL*cal[i]*ifac[i]%rhl; 70 getinv(b,invb,m),NTT(a,m<<1,1),NTT(invb,m<<1,1); 71 for (RG int i=0;i<(m<<1);++i) a[i]=1LL*a[i]*invb[i]%rhl; 72 NTT(a,m<<1,-1); printf("%lld\n",1LL*a[n]*fac[n-1]%rhl); 73 return 0; 74 }
bzoj3456 城市規劃