CodeForces 438E The Child and Binary Tree(DP + 生成函数 + 多项式模运算)

版权声明:本文为博主原创文章,转载请著名出处 http://blog.csdn.net/u013534123 https://blog.csdn.net/u013534123/article/details/85270223

大致题意:给定一个集合{Cn},一棵二叉树上的所有节点的点权值从这个集合中选取。现在给定一个m,问对于1..m中的每一个数字i,权值和恰好为i的不同的二叉树的个数有多少个。这里形态不同但点权集合的二叉树视为两种方案。

与前面做的题目类似,这种题目我们还是用dp的思维去考虑。令fi表示权值和为i的二叉树的个数。那么考虑增加一个点x,这个点的权值可以取i,Ci表示数值i在初始给定集合中是否出现过,两个儿子对应子树的点权值和为fj和fk,那么这样的话就会对点权和为ci*(fj+fk)的方案数产生贡献。于是我们可以写出状态转移方程:

                                                      \large f_x=\sum_{i=1}^{x}C_i\sum_{j=0}^{x-i}f_j*f_{x-i-j}

现在我们考虑这个式子。这是一个卷积套卷积的式子,直接去做的话是完成不了的。但是根据式子,我们表示成生成函数的形式,大致可以推出:

                                                                  \large f=f^2*C

然后根据C[0]=0,f[0]=1,可以完善,得到:

                                                             \large f=f^2*C+1

我们需要求f,于是把f当作位置数,这个就是一个一元二次方程。可以解得这个方程的根是:

                                                            \large f=\frac{1\pm \sqrt{1-4C}}{2}

进行分子有理化得:

                                                           \large f=\frac{2}{1\mp \sqrt{1-4C}}

考虑到当取0的时候,f=1,C=0,如果取减号,那么分母为0无意义,所有这里只能取加号。

                                                           \large f=\frac{2}{1 + \sqrt{1-4C}}

NTT加多项式开根再求逆元即可。具体见代码:

#include <bits/stdc++.h>
#define LL long long
using namespace std;

const int mod = 998244353;//(119 << 23) + 1;
const int modinv2 = (mod+1)/2; // 1/2 in F_p
const int G = 3;
const int N = 270010;

int c[N];

//取模加减乘
inline int add(int a,int b) {return a+b>=mod?a+b-mod:a+b;}
inline void inc(int&a,int b) {if ((a+=b)>=mod) a-=mod;}
inline int sub(int a,int b) {return a-b<0?a-b+mod:a-b;}
inline void dec(int&a,int b) {if ((a-=b)<0) a+=mod;}
inline int mul(int a,int b) {return (LL)a*b%mod;}
inline int qpow(int x,int n) {int ans=1;for (;n;n>>=1,x=(LL)x*x%mod) if (n&1) ans=(LL)ans*x%mod; return ans;}//quick power
//-------------------------------NTT--------------------------------
int wn[30],iwn[30]; //wn[i] = G^((P-1)/(2^i)) (mod P), iwn[i] = wn[i]^(-1) (mod P)
inline void init() //do this before NTT
{
    wn[23] = qpow(G,(mod-1)/(1<<23));
    for (int i=22;i>=0;i--) wn[i] = mul(wn[i+1],wn[i+1]);
    iwn[23] = qpow(wn[23],(1<<23)-1);
    for (int i=22;i>=0;i--) iwn[i] = mul(iwn[i+1],iwn[i+1]);
}
inline void revbin_permute(int a[],int n) {
    int i=1, j=n>>1, k;
    for (;i<n-1;i++) {
        if (i < j) swap(a[i],a[j]);
        for (k=n>>1;j>=k;) {j -= k; k >>= 1;}
        if (j < k) j += k;
    }
}

inline void NTT(int *f,int ldn,int is) {
    int n = (1<<ldn);
    revbin_permute(f,n);
    for (int i=0;i<n;i+=2) {
        int tmp1 = f[i], tmp2 = f[i+1];
        f[i] = add(tmp1,tmp2), f[i+1] = sub(tmp1,tmp2);
    }
    for (int ldm=2;ldm<=ldn;ldm++) {
        int m = (1<<ldm), mh = (m>>1);
        int dw = is>0?wn[ldm]:iwn[ldm], w = 1;
        for (int j=0;j<mh;j++) {
            for (int r=0;r<n;r+=m) {
                int u = f[r+j], v = mul(f[r+j+mh],w);
                f[r+j] = add(u,v);
                f[r+j+mh] = sub(u,v);
            }
            w = mul(w,dw);
        }
    }
}
//多项式乘法
inline void convolution(int *f,int *g,int n) {
    int ldn; for (int i=20;i>=0;i--) if (n&(1<<i)) {ldn=i;break;}
    NTT(f,ldn,1); NTT(g,ldn,1); //会改变g
    for (int i=0;i<n;i++) f[i] = mul(f[i],g[i]);
    NTT(f,ldn,-1);
    int iv = qpow(n,mod-2);
    for (int i=0;i<n;i++) f[i] = mul(f[i],iv);
}
//多项式求sq
inline void polysq(int *f,int n) {
    int ldn; for (int i=20;i>=0;i--) if (n&(1<<i)) {ldn=i;break;}
    NTT(f,ldn,1);
    for (int i=0;i<n;i++) f[i] = mul(f[i],f[i]);
    NTT(f,ldn,-1);
    int iv = qpow(n,mod-2);
    for (int i=0;i<n;i++) f[i] = mul(f[i],iv);
}
//多项式求inv
//Q(2n) = Q(n) - P*Q^2(n)
inline void polyinv(int *f,int n) {
    static int g[N],b[N],c[N];
    for (int i=0;i<n;i++) g[i]=0;
    g[0] = qpow(f[0],mod-2);
    for (int i=2;i<=n;i<<=1) {
        for (int j=0;j<i;j++) b[j] = g[j], c[j] = f[j];
        for (int j=i;j<2*i;j++) b[j] = c[j] = 0;
        polysq(b,2*i);
        for (int j=i;j<2*i;j++) b[j] = 0;
        convolution(b,c,2*i);
        for (int j=0;j<i;j++) g[j] = (2ll*g[j] - b[j] + mod)%mod;
    }
    for (int i=0;i<n;i++) f[i] = g[i];
}
//多项式求sqrt
//R(2n) = 1/2 * (R(n)+P(n)*R^{-1}(n))
inline void polysqrt(int *f,int n) {
    static int g[N],b[N],c[N];
    g[0] = 1; //根据需要改为sqrt(f[0])
    for (int i=2;i<=n;i<<=1) {
        for (int j=0;j<i;j++) b[j] = f[j], c[j] = g[j];
        for (int j=i;j<2*i;j++) b[j] = c[j] = 0;
        polyinv(c,i);
        convolution(b,c,2*i);
        for (int j=0;j<i;j++) g[j] = (LL)modinv2*(g[j] + b[j]) % mod;
    }
    for (int i=0;i<n;i++) f[i] = g[i];
}

int main()
{
    init(); int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        c[x]=mod-4;
    }
    int lg;
    for(lg=1;lg<=m;lg<<=1);
    c[0]++;polysqrt(c,lg);
    c[0]++; polyinv(c,lg);
    for(int i=1;i<=m;i++)
        printf("%d\n",(LL)2*c[i]%mod);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/u013534123/article/details/85270223