【bzoj3456】城市规划 【FFT/NTT】【多项式求逆】
题目大意:你需要求出n个点的简单(无重边无自环)无向连通图的数目模1004535809。。
题解:我们令表示n个点的简单无向图的数目,表示n个点的简单无向连通图的数目。
易得,意思是任意两点之间的边都可以取或不取。
同时,,意思是枚举点1所在连通块的大小i,从n-1个点里再选i-1个,连边组成一个联通块,剩下n-i个点随意连边。
我们变一下这个式子。
=>
=>
=>
可以发现这是一个卷积的形式。
我们令
,
则
=>
因此我们可以通过多项式求逆和多项式乘法得到B。
多项式求逆的关键公式:
最后的答案为。
代码
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=270005;
const ll mod=1004535809;
int n,m,rev[N];
ll jc[N],a[N],b[N],c[N],d[N];
ll fastpow(ll a,ll x){
a%=mod;
ll res=1;
while(x){
if(x&1){
res=res*a%mod;
}
x>>=1;
a=a*a%mod;
}
return res;
}
ll getinv(ll a){
return fastpow(a,mod-2);
}
void ntt(ll *a,int n,int dft){
for(int i=0;i<n;i++){
rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));
if(i<rev[i]){
swap(a[i],a[rev[i]]);
}
}
for(int i=1;i<n;i<<=1){
ll wn=fastpow(3,(mod-1)/i/2);
if(dft==-1){
wn=getinv(wn);
}
for(int j=0;j<n;j+=i<<1){
ll w=1,x,y;
for(int k=j;k<j+i;k++,w=w*wn%mod){
x=a[k];
y=w*a[k+i]%mod;
a[k]=(x+y)%mod;
a[k+i]=(x-y+mod)%mod;
}
}
}
if(dft==-1){
ll inv=getinv(n);
for(int i=0;i<n;i++){
a[i]=a[i]*inv%mod;
}
}
}
int main(){
scanf("%d",&m);
jc[0]=1;
for(int i=1;i<=m;i++){
jc[i]=jc[i-1]*i%mod;
}
for(n=1;n<=m;n<<=1);
a[0]=1;
for(int i=1;i<m;i++){
a[i]=fastpow(2,1LL*i*(i-1)/2)*getinv(jc[i]);
}
b[0]=getinv(a[0]);
for(int k=2;k<=n;k<<=1){
for(int i=0;i<k;i++){
c[i]=a[i];
if(i<(k>>1)){
d[i]=b[i];
}else{
d[i]=0;
}
}
ntt(c,k<<1,1);
ntt(d,k<<1,1);
for(int i=0;i<(k<<1);i++){
c[i]=(2*d[i]%mod-d[i]*d[i]%mod*c[i]%mod+mod)%mod;
}
ntt(c,k<<1,-1);
ntt(d,k<<1,-1);
for(int i=0;i<k;i++){
b[i]=c[i];
}
}
a[0]=0;
for(int i=1;i<=m;i++){
a[i]=fastpow(2,1LL*i*(i-1)/2)*getinv(jc[i-1]);
}
ntt(a,n<<1,1);
ntt(b,n<<1,1);
for(int i=0;i<(n<<1);i++){
a[i]=a[i]*b[i]%mod;
}
ntt(a,n<<1,-1);
printf("%lld\n",a[m]*jc[m-1]%mod);
return 0;
}