【刷题】LOJ 556 「Antileaf's Round」咱们去烧菜吧

题目描述

烧菜

你有 \(m\) 种物品,第 \(i\) 种物品的大小为 \(a_i\) ​,数量为 \(b_i\)​( \(b_i=0\) 表示有无限个)。

你还有 \(n\) 个背包,体积分别为 \(1\)\(n\) ,现在你很想知道用这些物品填满某个背包的方案数。

为了满足你的好奇心,你决定把填满每个背包的方案数都算一遍。

因为你其实只是闲得无聊,所以你只想知道方案数对 \(998244353\)\(7\times 17\times 2^{23}+1\),一个质数)取模后的值。

输入格式

第一行两个非负整数,分别表示 \(n,m\)

以下 \(m\) 行,每行两个非负整数,分别表示 \(a_i,b_i\) ​。

输出格式

输出 \(n\) 个非负整数表示答案。

样例

样例输入

5 3
1 0
1 1
3 2

样例输出

2
2
3
4
4

样例解释

拼出 \(1\) ~ \(5\) 的方案分别如下:

\(\{1_1\},\{1_2\}\)

\(\{1_1,1_1\},\{1_1,1_2\}\)

\(\{1_1,1_1,1_1\},\{1_1,1_1,1_2\},\{3\}\)

\(\{1_1,1_1,1_1,1_1\},\{1_1,1_1,1_1,1_2\},\{1_1,3\},\{1_2,3\}\)

\(\{1_1,1_1,1_1,1_1,1_1\},\{1_1,1_1,1_1,1_1,1_2\},\{1_1,1_1,3\},\{1_1,1_2,3\}\)

数据范围与提示

\(0<n,m \leq 10^5,0 \leq a_i​ \leq 110000,0 \leq b_i \leq 10^6\)

题解

数量无限的物品的生成函数为:\(\sum_{i=0}^{+\infty}x^{ia_i}=\frac{1}{1-x^{a_i}}\)

数量有限的物品的生成函数为:\(\sum_{i=0}^{b_i}x^{ia_i}=\frac{1-x^{(b_i+1)a_i}}{1-x^{a_i}}\)

将所有物品的生成函数乘起来就是答案的生成函数,即(假设全部都有限):

\(\begin{aligned} ans &= \prod_{i=1}^m\frac{1-x^{(b_i+1)a_i}}{1-x^{a_i}} \\ &= \exp{\sum_{i=1}^m (\ln (1-x^{(b_i+1)a_i})-\ln (1-x^{a_i}))} \end{aligned}\)

然后我们知道:\(-\ln(1-x)=\sum_{i=1}^{+\infty}\frac{x^i}{i}\) ,所以

\(\ln (1-x^{(b_i+1)a_i})=\sum_{i=1}^{+\infty}-\frac{1}{i}x^{(b_i+1)a_i}\)

\(\ln (1-x^{a_i})=\sum_{i=1}^{+\infty}-\frac{1}{i}x^{a_i}\)

那么对于物品,可以 \(O(n \ln{n} )\) 地把它们的贡献加到OGF上,然后做多项式exp就好了

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
#define ft first
#define sd second
#define pb(a) push_back(a)
#define PII std::pair<int,int>
#define PLL std::pair<ll,ll>
#define mp(a,b) std::make_pair(a,b)
#define ITR(a,b) for(auto a:b)
#define REP(a,b,c) for(register int a=(b),a##end=(c);a<=a##end;++a)
#define DEP(a,b,c) for(register int a=(b),a##end=(c);a>=a##end;--a)
const int MAXN=1<<21,Mod=998244353;
int M[MAXN];
ll OGF[MAXN],ans[MAXN];
template<typename T> inline void read(T &x)
{
    T data=0,w=1;
    char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
    if(ch!='\0')putchar(ch);
}
template<typename T> inline bool chkmin(T &x,T y){return y<x?(x=y,true):false;}
template<typename T> inline bool chkmax(T &x,T y){return y>x?(x=y,true):false;}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
namespace poly{
    const int Mod=998244353;
    int rev[MAXN];
    ll b[MAXN],c[MAXN],tmp[MAXN],inv2,inv3,invm,pw[MAXN],ipw[MAXN],iv[MAXN];
    inline ll qexp(ll a,ll b)
    {
        ll res=1;
        while(b)
        {
            if(b&1)res=res*a%Mod;
            a=a*a%Mod;
            b>>=1;
        }
        return res;
    }
    inline void init(int n)
    {
        inv2=qexp(2,Mod-2);inv3=qexp(3,Mod-2);
        for(register int l=1;l<=(n<<2);l<<=1)
            pw[l]=qexp(3,(Mod-1)/l),ipw[l]=qexp(inv3,(Mod-1)/l);
        REP(i,1,n+n+1)iv[i]=qexp(i,Mod-2);
    }
    inline void NTT(ll *A,int n,int tp)
    {
        REP(i,0,n-1)if(i<rev[i])std::swap(A[i],A[rev[i]]);
        for(register int l=2;l<=n;l<<=1)
        {
            ll wn=tp>0?pw[l]:ipw[l];
            for(register int i=0;i<n;i+=l)
            {
                ll w=1;
                for(register int j=0;j<(l>>1);++j)
                {
                    ll A1=A[i+j],A2=1ll*A[i+j+(l>>1)]*w%Mod;
                    A[i+j]=(A1+A2)%Mod,A[i+j+(l>>1)]=(A1-A2+Mod)%Mod;
                    w=1ll*w*wn%Mod;
                }
            }
        }
    }
    inline void Get_Mul(int n,ll *A,ll *B,ll *C)
    {
        int m,cnt;
        for(m=1,cnt=0;m<n;m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        NTT(A,m,1);NTT(B,m,1);
        REP(i,0,m-1)A[i]=1ll*A[i]*B[i]%Mod;
        NTT(A,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,n-1)C[i]=1ll*A[i]*invm%Mod;
    }
    inline void Get_Inv(int len,ll *A,ll *B)
    {
        if(len==1)
        {
            B[0]=qexp(A[0],Mod-2);
            return ;
        }
        Get_Inv((len+1)>>1,A,B);
        int m,cnt;
        for(m=1,cnt=0;m<(len<<1);m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        REP(i,0,len-1)tmp[i]=A[i];
        REP(i,len,m-1)tmp[i]=0;
        NTT(tmp,m,1);NTT(B,m,1);
        REP(i,0,m-1)B[i]=(2ll*B[i]%Mod-1ll*B[i]*B[i]%Mod*tmp[i]%Mod+Mod)%Mod;
        NTT(B,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,len-1)B[i]=1ll*B[i]*invm%Mod;
        REP(i,len,m-1)B[i]=0;
    }
    inline void Get_Sqr(int len,ll *A,ll *B)
    {
        if(len==1)
        {
            B[0]=1;
            return ;
        }
        Get_Sqr((len+1)>>1,A,B);
        memset(b,0,sizeof(b));Get_Inv(len,B,b);
        int m,cnt;
        for(m=1,cnt=0;m<(len<<1);m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        REP(i,0,len-1)tmp[i]=A[i];
        REP(i,len,m-1)tmp[i]=0;
        NTT(B,m,1);NTT(tmp,m,1);NTT(b,m,1);
        REP(i,0,m-1)B[i]=1ll*(tmp[i]+1ll*B[i]*B[i]%Mod)%Mod*b[i]%Mod*inv2%Mod;
        NTT(B,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,len-1)B[i]=1ll*B[i]*invm%Mod;
        REP(i,len,m-1)B[i]=0;
    }
    inline void Get_Div(int n1,ll *A,int n2,ll *B,ll *C,ll *D)
    {
        memset(c,0,sizeof(c));
        REP(i,0,n2-1)c[i]=B[n2-1-i];
        Get_Inv(n1-n2+1,c,b);
        int m,cnt;
        for(m=1,cnt=0;m<((n1-n2+1)<<1);m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        REP(i,0,n1-n2)tmp[i]=A[n1-1-i];
        REP(i,n1-n2+1,m-1)tmp[i]=0;
        NTT(tmp,m,1);NTT(b,m,1);
        REP(i,0,m-1)tmp[i]=1ll*tmp[i]*b[i]%Mod;
        NTT(tmp,m,-1);
        invm=qexp(m,Mod-2);
        memset(c,0,sizeof(c));
        REP(i,0,n1-n2)C[n1-n2-i]=c[n1-n2-i]=1ll*tmp[i]*invm%Mod;
        for(m=1,cnt=0;m<n1;m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        NTT(B,m,1);NTT(c,m,1);
        REP(i,0,m-1)B[i]=1ll*B[i]*c[i]%Mod;
        NTT(B,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,n2-2)D[i]=(A[i]-1ll*B[i]*invm%Mod+Mod)%Mod;
    }
    inline void Get_Der(int n,ll *A,ll *B)
    {
        REP(i,0,n-2)B[i]=1ll*A[i+1]*(i+1)%Mod;B[n-1]=0;
    }
    inline void Get_Int(int n,ll *A,ll *B)
    {
        REP(i,1,n-1)B[i]=1ll*A[i-1]*iv[i]%Mod;B[0]=0;
    }
    inline void Get_Ln(int n,ll *A,ll *B)
    {
        memset(b,0,sizeof(b));
        Get_Inv(n,A,b);Get_Der(n,A,c);
        int m,cnt;
        for(m=1,cnt=0;m<(n<<1);m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        NTT(b,m,1);NTT(c,m,1);
        REP(i,0,m-1)b[i]=1ll*b[i]*c[i]%Mod;
        NTT(b,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,n-1)b[i]=1ll*b[i]*invm%Mod;
        Get_Int(n,b,B);
    }
    inline void Get_Exp(int len,ll *A,ll *B)
    {
        if(len==1)
        {
            B[0]=1;
            return ;
        }
        Get_Exp((len+1)>>1,A,B);Get_Ln(len,B,c);
        int m,cnt;
        for(m=1,cnt=0;m<(len<<1);m<<=1,++cnt);
        REP(i,0,m-1)rev[i]=(rev[i>>1]>>1)|((i&1)<<(cnt-1));
        REP(i,0,len-1)tmp[i]=(A[i]-c[i]+(i==0)+Mod)%Mod;
        REP(i,len,m-1)tmp[i]=0;
        NTT(tmp,m,1);NTT(B,m,1);
        REP(i,0,m-1)B[i]=1ll*tmp[i]*B[i]%Mod;
        NTT(B,m,-1);
        invm=qexp(m,Mod-2);
        REP(i,0,len-1)B[i]=1ll*B[i]*invm%Mod;
        REP(i,len,m-1)B[i]=0;
    }
}
int main()
{
    int n,m;read(n);read(m);poly::init(n+1);
    REP(i,1,m)
    {
        int a,b;read(a);read(b);
        if(a<=n)M[a]++;
        if(1ll*(b+1)*a<=n&&b)M[(b+1)*a]--;
    }
    REP(i,1,n)if(M[i])REP(j,1,n/i)OGF[i*j]=((OGF[i*j]+1ll*M[i]*poly::iv[j]%Mod)%Mod+Mod)%Mod;
    poly::Get_Exp(n+1,OGF,ans);
    REP(i,1,n)write(ans[i],'\n');
    return 0;
}

转载于:https://www.cnblogs.com/hongyj/p/10518744.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值