【BZOJ-3456】城市规划 CDQ分治 + NTT

题目链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3456

Solution

这个问题可以考虑dp，利用补集思想

N个点的简单图总数量为$2^{\binom{N}{2}}$，要求的是简单联通图，所以可以用总量减不连通的。

不连通的可以通过枚举与某个固定点的联通的点的数量得到$tot=\sum _{i=1} ^{N} \binom{N-1}{i-1}*dp[i]*2^{\binom{N-i}{2}}$

其中$dp[i]$表示的就是$i$个点的联通图数量。

然后将公式稍稍变型整理成$\frac{dp[N]}{(N-1)!}=\frac{2^{\binom{N}{2}}}{(N-1)!}-\sum_{i=1}^{N-1}\frac{dp[i]}{(i-1)!}*\frac{2^{\binom{N-i}{2}}}{(N-i)!}$

这个式子可以利用 CDQ分治+NTT 在$O(Nlog^{2}N)$的时间得到。

至于这道题吗，显然是可以多项式求逆来做的，复杂度$O(NlnN)$，上述做法自己写的被卡常了，不过本机效果还不错，留下代码以后看看。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define LL long long#define P 1004535809LL
#define G 3LL#define MAXN 800010int N,len;inline LL Pow(LL x,LL y) {LL re=1; for (LL i=y; i; i>>=1,x=x*x%P) if (i&1) re=re*x%P; return re;}inline LL Inv(LL x) {return Pow(x,P-2);}int A[MAXN],B[MAXN],ans[MAXN],wn[31],dp[MAXN];inline void Rader(int *x)
{for (register int i=1,j=len>>1,k; i<len-1; i++) {if (i<j) swap(x[i],x[j]);k=len>>1;while (j>=k) j-=k,k>>=1;if (j<k) j+=k;}
}inline void DFT(int *x,int opt)
{Rader(x);for (register int h=2,id=0; h<=len; h<<=1) {LL Wn=wn[++id];for (register int i=0; i<len; i+=h) {LL W=1;for (register int j=i; j<i+h/2; j++) {LL u=(LL)x[j]%P,t=(LL)W*x[j+h/2]%P;x[j]=(u+t)%P; x[j+h/2]=(u-t+P)%P;W=W*Wn%P;}}}if (opt==-1) {for (register int i=1; i<len/2; i++) swap(x[i],x[len-i]);for (register int i=0; i<len; i++) x[i]=(LL)x[i]*Inv(len)%P;}
}inline void NTT()
{DFT(A,1); DFT(B,1);for (register int i=0; i<len; i++) ans[i]=(LL)A[i]*B[i]%P;DFT(ans,-1);
}int C2[MAXN],fac[MAXN],ifac[MAXN];inline void CDQ(int l,int r)
{if (l==r) {dp[l]=(C2[l]-(LL)dp[l]*fac[l-1]%P+P)%P; return;}int mid=(l+r)>>1;CDQ(l,mid);for (register int i=l; i<=mid; i++) A[i-l]=(LL)dp[i]*ifac[i-1]%P;for (register int i=0; i<=r-l; i++) B[i]=(LL)C2[i]*ifac[i]%P;for (register int i=mid-l+1; i<=r-l; i++) A[i]=0;len=1; while (len<((r-l+1)<<1)) len<<=1;for (register int i=r-l+1; i<len; i++) A[i]=B[i]=0;NTT();for (register int i=mid+1; i<=r; i++) (dp[i]+=ans[i-l])%=P;CDQ(mid+1,r);
}int main()
{
//  freopen("count.in","r",stdin);
//  freopen("count.out","w",stdout);scanf("%d",&N);for (register int i=0; i<=30; i++) wn[i]=Pow(G,(P-1)/(1<<i));fac[0]=ifac[0]=1;for (register int i=1; i<=N; i++) fac[i]=((LL)fac[i-1]*i)%P,ifac[i]=Inv(fac[i]);for (register int i=1; i<=N; i++) C2[i]=Pow(2LL,(LL)((LL)i*(i-1))/2);CDQ(1,N);printf("%d\n",dp[N]);return 0;
}

转载于:https://www.cnblogs.com/DaD3zZ-Beyonder/p/6544454.html