洛谷 P4841 城市规划 ntt

最新推荐文章于 2021-04-14 21:51:19 发布

Amber_lylovely

最新推荐文章于 2021-04-14 21:51:19 发布

阅读量255

点赞数

分类专栏： fft-ntt

本文链接：https://blog.csdn.net/liangzihao1/article/details/82977520

版权

fft-ntt 专栏收录该内容

27 篇文章 0 订阅

订阅专栏

题目描述

刚刚解决完电力网络的问题, 阿狸又被领导的任务给难住了.
刚才说过, 阿狸的国家有n个城市, 现在国家需要在某些城市对之间建立一些贸易路线, 使得整个国家的任意两个城市都直接或间接的连通.
为了省钱, 每两个城市之间最多只能有一条直接的贸易路径. 对于两个建立路线的方案, 如果存在一个城市对, 在两个方案中是否建立路线不一样, 那么这两个方案就是不同的, 否则就是相同的. 现在你需要求出一共有多少不同的方案.
好了, 这就是困扰阿狸的问题. 换句话说, 你需要求出n个点的简单(无重边无自环)无向连通图数目.
由于这个数字可能非常大, 你只需要输出方案数mod 1004535809(479 * 2 ^ 21 + 1)即可.

输入输出格式

输入格式：
仅一行一个整数n(<=130000)

输出格式：
仅一行一个整数, 为方案数 mod 1004535809.

输入输出样例

输入样例#1：
3
输出样例#1：
4
输入样例#2：
4
输出样例#2：
38
输入样例#3：
100000
输出样例#3：
829847355
说明

对于 20%的数据, n <= 10
对于 40%的数据, n <= 1000
对于 60%的数据, n <= 30000
对于 80%的数据, n <= 60000
对于 100%的数据, n <= 130000
试题来源：2013中国国家集训队第二次作业

分析：
无向图最多有 $\frac{n*(n-1)}{2}$ 条边，考虑每条边是否出现可以得出，n个点的无向图有 $2^{\frac{n*(n-1)} {2}}$ 个。
设 $f (i)$ 为 $i$ 个点连通的无向图数， $g (i)$ 为总的连通图数。
考虑1号点的连通块大小。假设连通块大小为 $j$ ，那么相当于要从 $i - 1$ 个点中找出 $j - 1$ 个点，有 $\binom{i-1}{j-1}$ 种方案，而这些点连通有 $f (j)$ 种方案。对于剩下的点可以乱连，但是不能连向这 $j$ 个点，也就是 $i - j$ 个点的无向图数，也就是 $g (i - j)$ 。
所以，
$g(i)=\sum_{j=1}^{i}f(j)*\binom{i-1}{j-1}*g(i-j)$
化一下式子就是，
$\frac{g(i)}{(i-1)!}=\sum_{j=1}^{i}\frac{f(j)}{(j-1)!}*\frac{g(i-j)}{(i-j)!}$
这个就是一个卷积形式，其中 $g(i)=2^{\frac{i*(i-1)}{2}}$ 。
设 $F(i)=\frac{f(i)}{(i-1)!}$ ， $G(i)=\frac{g(i)}{i!}$ ， $H(i)=\frac{g(i)}{(i-1)!}$
那么 $H = F * G$ ，所以 $F=H*G^{-1}$ ，直接求逆即可。

代码：

// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long

const int maxn=3e5+7;
const LL mod=1004535809;
const LL G=3;

using namespace std;

int n,len,r[maxn];
LL f[maxn],g[maxn],a[maxn],b[maxn],c[maxn],inv[maxn],jc[maxn],w[maxn];

LL power(LL x,LL y)
{
    if (!y) return 1;
    LL c=power(x,y/2);
    c=(c*c)%mod;
    if (y%2) c=(c*x)%mod;
    return c;
}

void ntt(LL *a,int f)
{
    for (int i=0;i<len;i++)
    {
        if (i<r[i]) swap(a[i],a[r[i]]);
    }
    w[0]=1;
    for (int i=2;i<=len;i*=2)
    {
        LL wn;
        if (f==1) wn=power(G,(LL)(mod-1)/i);
             else wn=power(G,(LL)(mod-1)-(mod-1)/i);
        for (int j=i/2;j>=0;j-=2) w[j]=w[j/2];
        for (int j=1;j<i/2;j+=2) w[j]=(w[j-1]*wn)%mod;
        for (int j=0;j<len;j+=i)
        {
            for (int k=0;k<i/2;k++)
            {
                LL u=a[j+k],v=a[j+k+i/2]*w[k]%mod;
                a[j+k]=(u+v)%mod;
                a[j+k+i/2]=(u+mod-v)%mod;
            }
        }
    }
    if (f==-1)
    {
        LL inv=power(len,mod-2);
        for (int i=0;i<len;i++) a[i]=a[i]*inv%mod;
    }
} 

void NTT(LL *f,LL *g,LL *h,int n,int m)
{
    len=1;
    int k=0;
    while (len<=(n+m)) len<<=1,k++;
    for (int i=0;i<len;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(k-1));
    for (int i=0;i<len;i++)
    {
        if (i<n) a[i]=f[i]; 
            else a[i]=0;
        if (i<m) b[i]=g[i];
            else b[i]=0;
    }
    ntt(a,1); ntt(b,1);
    for (int i=0;i<len;i++) h[i]=a[i]*b[i]%mod;
    ntt(h,-1);
}

void getinv(LL *f,LL *g,int deg)
{
    if (deg==1)
    {
        g[0]=power(f[0],mod-2);
        return;
    }
    int mid=(deg+1)/2;
    getinv(f,g,mid);
    NTT(f,g,c,deg,mid);
    c[0]=(2+mod-c[0])%mod;
    for (int i=1;i<deg;i++) c[i]=(mod-c[i])%mod;	
    NTT(c,g,g,deg,mid);
}

int main()
{
    scanf("%d",&n);
    jc[0]=1;
    for (LL i=1;i<=n;i++) jc[i]=(jc[i-1]*i)%mod;
    for (LL i=1;i<=n;i++) f[i]=power(2,(i*(i-1)/2)%(mod-1))*power(jc[i-1],mod-2)%mod;
    for (LL i=1;i<=n;i++) g[i]=f[i]*power(i,mod-2)%mod;
    g[0]=1;	
    getinv(g,inv,n+1);
    NTT(f,inv,f,n+1,n+1);
    printf("%lld",f[n]*jc[n-1]%mod);
}