指数型生成函数&[bzoj3456]城市规划

最新推荐文章于 2019-02-22 21:30:55 发布

zhouyuheng2003

最新推荐文章于 2019-02-22 21:30:55 发布

阅读量405

点赞数 1

分类专栏： OI NTT 生成函数多项式ln

本文链接：https://blog.csdn.net/zhouyuheng2003/article/details/86692330

版权

OI 同时被 3 个专栏收录

104 篇文章 2 订阅

订阅专栏

NTT

11 篇文章 0 订阅

订阅专栏

生成函数

1 篇文章 0 订阅

订阅专栏

前言

~~打完多项式板子后的第一题+清真的题意~~

题目相关

题目链接

题目大意

求 $n$ 个点的简单(无重边无自环)无向连通图数目
输出模 $1004535809(479*2^{21}+1)$

数据范围

$n\le 130000$

题解

暴力

我只推出了个暴力，~~感觉和多项式没啥关系~~
如果没有任何限制，那么答案为 $2^{C_n^2}$
设 $F_n$ 为答案，写出递推式（用所有的方案数减去不连通的方案数，枚举最后一个点所处的联通块大小即可）
$F_i=2^{C_i^2}-\sum_{j=1}^{i-1}F_j*C_{i-1}^{j-1}*2^{C_{i-j}^2}$
递推+快速幂
复杂度 $\mathcal O(n^2logp)$
~~要是没有代码长度限制就可以打表过了（大雾）~~

正解

~~和暴力几乎没啥关系~~
考虑指数型生成函数！
我们来观察指数型生成函数的性质，列出两个指数型生成函数
$A(x)=\sum_{i=0}^{\infty}a_i\frac{x^i}{i!}$
$B(x)=\sum_{i=0}^{\infty}b_i\frac{x^i}{i!}$
我们观察一下 $A (x) \cdot B (x)$
$\begin{aligned} A(x)·B(X)&=(\sum_{i=0}^{\infty}a_i\frac{x^i}{i!})·(\sum_{i=0}^{\infty}b_i\frac{x^i}{i!})\\ &=\sum_{i=0}^{\infty}a_i\frac{x^i}{i!}\sum_{j=0}^{\infty}b_j\frac{x^j}{j!}\\ &=\sum_{i=0}^{\infty}\sum_{j=0}^{\infty}a_i\frac{x^i}{i!}b_j\frac{x^j}{j!}\\ &=\sum_{n=0}^{\infty}(\sum_{i=0}^{\infty}\binom{n}{i}a_ib_{n-i})\frac{x^n}{n!}\\ \end{aligned}$
发现这个形式非常的奇妙
如果 $a_i$ 代表在第一个集合中选 $i$ 个的方案数， $b_i$ 代表在第二个集合中选 $i$ 个的方案数
那么相乘后我们就可以知道总共选 $i$ 个的方案数

考虑划分
我们发现，如果我们按照划分出一个大小为 $i$ 的集合的方案数建一个指数型生成函数 $F (x)$
那么我们就能计算出对于 $n$ 个元素划分成 $m$ 个集合的方案数
对于 $m = 0$ ，答案的多项式为 $1$ （即只有 $n = 0$ 时方案数为 $1$ ）
对于 $m = 1$ ，答案多项式为 $\frac{F(x)}{1!}$
对于 $m = 2$ ，答案多项式为 $\frac{F^2(x)}{2!}$
对于 $m = 3$ ，答案多项式为 $\frac{F^3(x)}{3!}$
···
由于划分先后的问题，所以要除以阶乘
如果我们要求每个 $n$ 的划分总方案数
即 $\frac{F^0(x)}{0!}+\frac{F^1(x)}{1!}+\frac{F^(x)}{2!}···=\sum_{i=0}^\infty \frac{F^i(x)}{i!}$
根据底数为 $e$ 的泰勒展开式 $e^x=\sum_{i=0}^\infty\frac {x^i}{i!}$
我们得到 $\sum_{i=0}^\infty \frac{F^i(x)}{i!}=e^{F(x)}$

回到本题： 我们思考，在这一题中，所有联通图都是无限制图的一个划分
设 $F (x)$ 为联通图方案数的多项式， $G (x)$ 为无限制图方案数的多项式
容易发现 $e^{F(x)}=G(x)$
所以 $F (x) = l n (G (x))$
我们发现 $G (x)$ 非常好求（暴力的那部分已经说过了）
然后直接多项式ln即可

代码

~~拉了板子所以代码很长~~

#include<cstdio>
#include<cctype>
#include<cstring>
#include<cstdlib>
#include<vector>
namespace fast_IO
{
    const int IN_LEN=10000000,OUT_LEN=10000000;
    char ibuf[IN_LEN],obuf[OUT_LEN],*ih=ibuf+IN_LEN,*oh=obuf,*lastin=ibuf+IN_LEN,*lastout=obuf+OUT_LEN-1;
    inline char getchar_(){return (ih==lastin)&&(lastin=(ih=ibuf)+fread(ibuf,1,IN_LEN,stdin),ih==lastin)?EOF:*ih++;}
    inline void putchar_(const char x){if(oh==lastout)fwrite(obuf,1,oh-obuf,stdout),oh=obuf;*oh++=x;}
    inline void flush(){fwrite(obuf,1,oh-obuf,stdout);}
}
using namespace fast_IO;
#define getchar() getchar_()
#define putchar(x) putchar_((x))
typedef long long ll;
#define rg register
template <typename T> inline T max(const T a,const T b){return a>b?a:b;}
template <typename T> inline T min(const T a,const T b){return a<b?a:b;}
template <typename T> inline T mind(T&a,const T b){a=a<b?a:b;}
template <typename T> inline T maxd(T&a,const T b){a=a>b?a:b;}
template <typename T> inline T abs(const T a){return a>0?a:-a;}
template <typename T> inline void swap(T&a,T&b){T c=a;a=b;b=c;}
template <typename T> inline void swap(T*a,T*b){T c=a;a=b;b=c;}
template <typename T> inline T gcd(const T a,const T b){if(!b)return a;return gcd(b,a%b);}
template <typename T> inline T square(const T x){return x*x;};
template <typename T> inline void read(T&x)
{
    char cu=getchar();x=0;bool fla=0;
    while(!isdigit(cu)){if(cu=='-')fla=1;cu=getchar();}
    while(isdigit(cu))x=x*10+cu-'0',cu=getchar();
    if(fla)x=-x;  
}
template <typename T> void printe(const T x)
{
    if(x>=10)printe(x/10);
    putchar(x%10+'0');
}
template <typename T> inline void print(const T x)
{
    if(x<0)putchar('-'),printe(-x);
    else printe(x);
}
const int maxn=2097152,mod=1004535809;
inline int Md(const int x){return x>=mod?x-mod:x;}
template<typename T>
inline int pow(int x,T y)
{
    rg int res=1;x%=mod;
    for(;y;y>>=1,x=(ll)x*x%mod)if(y&1)res=(ll)res*x%mod;
    return res;
}
namespace Poly///namespace of Poly
{
int W_[maxn],FW_[maxn],ha[maxn],hb[maxn],Inv[maxn];
inline void init(const int x)
{
    rg int tim=0,lenth=1;
    while(lenth<x)lenth<<=1,tim++;
    for(rg int i=1;i<lenth;i++)
    {
        W_[i]=pow(3,(mod-1)/i/2);
        FW_[i]=pow(W_[i],mod-2);
    }
    Inv[0]=Inv[1]=1;
    for(rg int i=2;i<x;i++)Inv[i]=(ll)(mod-mod/i)*Inv[mod%i]%mod;
}
int L;
inline void NTT(int*A,const int fla)//prepare:init L 
{
    for(rg int i=0,j=0;i<L;i++)
    {
        if(i>j)swap(A[i],A[j]);
        for(rg int k=L>>1;(j^=k)<k;k>>=1);
    }
    for(rg int i=1;i<L;i<<=1)
    {
        const int w=fla==-1?FW_[i]:W_[i];
        for(rg int j=0,J=i<<1;j<L;j+=J)
        {
            int K=1;
            for(rg int k=0;k<i;k++,K=(ll)K*w%mod)
            {
                const int x=A[j+k],y=(ll)A[j+k+i]*K%mod;
                A[j+k]=Md(x+y),A[j+k+i]=Md(mod+x-y);
            }
        }
    }
}
inline int Quadratic_residue(const int a)
{
	if(a==0)return 0;
	int b=(rand()<<14^rand())%mod;
	while(pow(b,(mod-1)>>1)!=mod-1)b=(rand()<<14^rand())%mod;
	int s=mod-1,t=0,x,inv=pow(a,mod-2),f=1;
	while(!(s&1))s>>=1,t++,f<<=1;
	t--,x=pow(a,(s+1)>>1),f>>=1;
	while(t)
	{
		f>>=1;
		if(pow((int)((ll)inv*x%mod*x%mod),f)!=1)x=(ll)x*pow(b,s)%mod;
		t--,s<<=1;
	}
	return min(x,mod-x);
}
struct poly
{
    std::vector<int>A;
    poly(){A.resize(0);}
    poly(const int x){A.resize(1),A[0]=x;}
    inline int&operator[](const int x){return A[x];}
    inline int operator[](const int x)const{return A[x];}
    inline void clear(){A.clear();}
    inline unsigned int size()const{return A.size();}
    inline void resize(const unsigned int x){A.resize(x);}
    void RE(const int x)
    {
        A.resize(x);
        for(rg int i=0;i<x;i++)A[i]=0; 
    }
    void readin(const int MAX)
    {
        A.resize(MAX);
        for(rg int i=0;i<MAX;i++)read(A[i]);
    }
    void putout()const
    {
        for(rg unsigned int i=0;i<A.size();i++)print(A[i]),putchar(' ');
    }
    inline poly operator +(const poly b)const
    {
        poly RES;
        RES.resize(max(size(),b.size()));
        for(rg unsigned int i=0;i<RES.size();i++)RES[i]=Md((i<size()?A[i]:0)+(i<b.size()?b[i]:0));
        return RES;
    }
    inline poly operator -(const poly b)const
    {
        poly RES;
        RES.resize(max(size(),b.size()));
        for(rg unsigned int i=0;i<RES.size();i++)RES[i]=Md((i<size()?A[i]:0)+mod-(i<b.size()?b[i]:0));
        return RES;
    }
    inline poly operator *(const int b)const
	{
		poly RES=*this;
        for(rg unsigned int i=0;i<RES.size();i++)RES[i]=(ll)RES[i]*b%mod;
        return RES;
	}
	inline poly operator /(const int b)const
	{
		poly RES=(*this)*pow(b,mod-2);
    	return RES;
	}
    inline poly operator *(const poly b)const
    {
        const int RES=A.size()+b.size()+1;
        L=1;while(L<RES)L<<=1;
        poly c;c.A.resize(RES);
        memset(ha,0,sizeof(int)*L);
        memset(hb,0,sizeof(int)*L);
        for(rg unsigned int i=0;i<A.size();i++)ha[i]=A[i];
        for(rg unsigned int i=0;i<b.A.size();i++)hb[i]=b.A[i];
        NTT(ha,1),NTT(hb,1);
        for(rg int i=0;i<L;i++)ha[i]=(ll)ha[i]*hb[i]%mod;
        NTT(ha,-1);
        const int inv=pow(L,mod-2);
        for(rg int i=0;i<RES;i++)c.A[i]=(ll)ha[i]*inv%mod;
        return c;
    }
    inline poly inv()const
    {
        poly C;
        if(A.size()==1){C=*this;C[0]=pow(C[0],mod-2);return C;}
        C.resize((A.size()+1)>>1);
        for(rg unsigned int i=0;i<C.size();i++)C[i]=A[i];
        C=C.inv();
        L=1;while(L<(int)size()*2-1)L<<=1;
        for(rg unsigned int i=0;i<A.size();i++)ha[i]=A[i];
        for(rg unsigned int i=0;i<C.size();i++)hb[i]=C[i];
        memset(ha+A.size(),0,sizeof(int)*(L-A.size()));
        memset(hb+C.size(),0,sizeof(int)*(L-C.size()));
        NTT(ha,1),NTT(hb,1);
        for(rg int i=0;i<L;i++)ha[i]=(ll)(2-(ll)hb[i]*ha[i]%mod+mod)*hb[i]%mod;
        NTT(ha,-1);
        const int inv=pow(L,mod-2);
        C.resize(size());
        for(rg unsigned int i=0;i<size();i++)C[i]=(ll)ha[i]*inv%mod;
        return C;
    }
/*    inline poly inv()const
    {
        poly C;
        if(A.size()==1){C=*this;C[0]=pow(C[0],mod-2);return C;}
        C.resize((A.size()+1)>>1);
        for(rg unsigned int i=0;i<C.size();i++)C[i]=A[i];
        C=C.inv();
        poly D=(poly)2-*this*C;
        D.resize(size());
        D=D*C;
        D.resize(size());
        return D;
    }*///大常数版本 
    inline void Reverse(const int n)
    {
    	A.resize(n);
    	for(rg int i=0,j=n-1;i<j;i++,j--)swap(A[i],A[j]);
    }
    inline poly operator /(const poly B)const
	{
		poly a=*this,b=B;a.Reverse(size()),b.Reverse(B.size());
		b.resize(size()-B.size()+1);
		b=b.inv();
		b=b*a;
		b.Reverse(size()-B.size()+1);
		return b;
	}
	inline poly operator %(const poly B)const
	{
		poly C=(*this)-(*this)/B*B;C.resize(B.size()-1);
		return C;
	}
	inline poly diff()const
	{
		poly C;C.resize(size()-1);
		for(rg unsigned int i=1;i<size();i++)C[i-1]=(ll)A[i]*i%mod;
		return C;
	}
	inline poly inte()const
	{
		poly C;C.resize(size()+1);
		for(rg unsigned int i=0;i<size();i++)C[i+1]=(ll)A[i]*Inv[i+1]%mod;
		C[0]=0;
		return C;
	}
	inline poly ln()const
	{
		poly C=(diff()*inv()).inte();C.resize(size());
		return C;
	}
	inline poly exp()const
	{
		poly C;
		if(size()==1){C=*this;C[0]=1;return C;}
		C.resize((size()+1)>>1);
		for(rg unsigned int i=0;i<C.size();i++)C[i]=A[i];
		C=C.exp();C.resize(size());
		poly D=(poly)1-C.ln()+*this;
		D=D*C;
		D.resize(size());
		return D;
	}
	inline poly sqrt()const
	{
		poly C;
		if(size()==1)
		{
			C=*this;C[0]=Quadratic_residue(C[0]);
			return C;
		}
		C.resize((size()+1)>>1);
		for(rg unsigned int i=0;i<C.size();i++)C[i]=A[i];
		C=C.sqrt();C.resize(size());
		C=(C+*this*C.inv())*(int)499122177;
		C.resize(size());
		return C;
	}
	inline poly operator >>(const unsigned int x)const
    {
    	poly C;if(size()<x){C.resize(0);return C;}
		C.resize(size()-x);
    	for(rg unsigned int i=0;i<C.size();i++)C[i]=A[i+x];
    	return C;
    }
    inline poly operator <<(const unsigned int x)const
    {
    	poly C;C.RE(size()+x);
    	for(rg unsigned int i=0;i<size();i++)C[i+x]=A[i];
    	return C;
    }
    inline poly Pow(const unsigned int x)const
    {
    	for(rg unsigned int i=0;i<size();i++)
			if(A[i])
			{
				poly C=((((*this/A[i])>>i).ln()*x).exp()*pow(A[i],x))<<(min(i*x,size()));
				C.resize(size());
				return C;
			}
    	return *this;
    }
};
}///namespace of Poly
Poly::poly a;
int n;
int fac[maxn+1],inv[maxn+1]; 
int main()
{
	fac[0]=1;for(rg int i=1;i<=maxn;i++)fac[i]=(ll)fac[i-1]*i%mod;
	inv[maxn]=pow(fac[maxn],mod-2);for(rg int i=maxn;i>=1;i--)inv[i-1]=(ll)inv[i]*i%mod;
    Poly::init(maxn);///namespace of Poly
    read(n);
    a.RE(n+1);
	for(rg int i=0;i<=n;i++)a[i]=(ll)pow(2,((ll)i*(i-1))>>1)*inv[i]%mod;
	a=a.ln();
	print((ll)a[n]*fac[n]%mod);
	return flush(),0;
}

总结

好推+好用的指数型生成函数！

zhouyuheng2003

关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
指数型生成函数&[bzoj3456]城市规划

前言打完多项式板子后的第一题+清真的题意题目相关题目链接题目大意求nnn个点的简单(无重边无自环)无向连通图数目输出模1004535809(479∗221+1)1004535809(479*2^{21}+1)1004535809(479∗221+1)数据范围n≤130000n\le 130000n≤130000题解暴力我只推出了个暴力，感觉和多项式没啥关系如果没有任何限制，...
复制链接

扫一扫

专栏目录