BZOJ3027：Sweet/BZOJ1272：Gate Of Babylon（生成函数+广义二项式定理+卢卡斯定理）

最新推荐文章于 2021-02-13 17:09:56 发布

Fat_tu

最新推荐文章于 2021-02-13 17:09:56 发布

阅读量311

点赞数

分类专栏：生成函数

本文链接：https://blog.csdn.net/q582116859/article/details/79547270

版权

生成函数专栏收录该内容

12 篇文章 0 订阅

订阅专栏

题面
首先问题可以差分

对于每一罐都搞一个生成函数
大概是 $\frac{1-x^{m+1}}{1-x}$
总方案数的生成函数就是

1 ( 1 - x ) n \prod i = 1 n (1 - m i)

$\frac{1}{(1-x)^n}\prod_{i=1}^{n}(1-m_i)$

= (1 - x) - n \prod i = 1 n (1 - m i)

$=(1-x)^{-n}\prod_{i=1}^{n}(1-m_i)$

传说中的性质

\sum i = 0 o o C i n + i - 1 x i = (1 - x) - n

$\sum_{i=0}^{oo}C_{n+i-1}^ix^i=(1-x)^{-n}$
展开
由于n比较小，我们可以枚举

∏ni=1(1−mi) ∏ i = 1 n ( 1 − m i ) $\prod_{i=1}^{n}(1-m_i)$ 的每项系数f
求

f[k] f [ k ] $f[k]$ 的贡献，大概为

\sum i = 0 b - k C n - 1 n - i - 1 = C k n + m - k

$\sum_{i=0}^{b-k}C_{n-i-1}^{n-1}=C_{n+m-k}^{k}$

两题几乎一样
BZOJ1272要用卢卡斯定理

bzoj3027

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=2020;
const LL p=2004;

LL mo=2004,jc=1;
LL n,nn,a,b,m[N];

LL C(LL x,LL y)
{
    if(x<y)
    return 0;
    LL res=1;
    for(int i=x;i>=x-y+1;i--)
    res=res*i%mo;
    return (res/jc)%p;
}

LL work(LL x)
{
    LL res=0;
    for(int i=0;i<=nn;i++)
    {
        LL s=x,ops=1;
        for(int j=1;j<=n;j++)
        if((1<<(j-1))&i)
        ops=-ops,s-=m[j]+1;
        if(s<0)
        continue;
        res+=C(s+n,n)*ops;
    }
    return res%p;
}

int main()
{
    cin>>n>>a>>b;;
    nn=(1<<n)-1;

    for(int i=2;i<=n;i++)
    jc*=i,mo*=i;

    for(int i=1;i<=n;i++)
    cin>>m[i];
    cout<<((work(b)-work(a-1))%p+p)%p<<endl;

    return 0;
}

bzoj1272

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=1234567;

LL n,T,p,m,nn,ans;
LL b[N];
LL jc[N],Ijc[N],I[N];

LL C(LL x,LL y)
{
    if(x<y)
    return 0;
    if(x<p)
    return jc[x]*Ijc[y]%p*Ijc[x-y]%p;
    return C(x/p,y/p)*C(x%p,y%p)%p; 
}

int main()
{
    cin>>n>>T>>m>>p;
    nn=(1<<T)-1;

    for(int j=1;j<=T;j++)
    scanf("%d",&b[j]);

    I[1]=Ijc[0]=jc[0]=1;
    for(int i=2;i<p;i++)
    I[i]=I[p%i]*(p-p/i)%p;

    for(int i=1;i<p;i++)
    jc[i]=jc[i-1]*i%p,Ijc[i]=Ijc[i-1]*I[i]%p;

    for(int i=0;i<=nn;i++)
    {
        LL ops=1,s=m;
        for(int j=1;j<=T;j++)
        if(i&(1<<(j-1)))
        ops=-ops,s-=b[j]+1;
        if(s<0)
        continue;
        ans=(ans+ops*C(n+s,n)+p)%p;
    }

    cout<<ans<<endl;

    return 0;
}