容斥原理+快速幂求逆元+lucas

最新推荐文章于 2024-01-21 17:26:58 发布

Togt

最新推荐文章于 2024-01-21 17:26:58 发布

阅读量271

点赞数 1

分类专栏：容斥原理数论文章标签： cf

本文链接：https://blog.csdn.net/qq_35160381/article/details/80260360

版权

数论同时被 2 个专栏收录

9 篇文章 0 订阅

订阅专栏

容斥原理

1 篇文章 0 订阅

订阅专栏

快速幂

ll qm(ll a,ll n)
{
    ll ans=1;
    if(a>=mod)a%=mod;//尤其当a可能爆int时注意先加一个mod
    while(n)
    {
        if(n&1)
        {
            ans=ans*a%mod;
        }
        n>>=1;
        a=a*a%mod;
    }
    return ans;
}

逆元：inv[i]=qm(i,mod-2);

lucas定理：

设：a>b，且 $a=a_0+a_1*q^1+...+a_k*q^n$ ， $b=b_0+b_1*q^1+...+b_k*q^n$

则有 $C^b_a=C^{b_0}_{a_0}C^{b_1}_{a_1}...C^{b_k}_{a_k}$

在组合数中规定，若 $a<b$ ，那么 $C^b_a=0$

结论：如果 $C^b_a$ 要为奇数，则a的每一位必定不小于b，否则出现一项为0，乘积为0，即为偶数；简单判定：若a&b=b，则 $C^b_a$ 为奇数。

简单组合数的实现：

ll C(ll n, ll m)//由于m<=20可以最直接求C(n,m)
{
    if(n<m)return 0;
    ll ans=1;
    for(int i=1;i<=m;i++)
    {
        ans = (n-i+1)%mod*ans%mod*inv[i]%mod;//long long 面前注意写mod，否则可能wa到怀疑人生
    }
    return ans;
}

lucas:

ll lucas(ll n,ll m)//适合p较小
{
    if(m==0)return 1;
    return C(n%mod,m%mod)*lucas(n/mod,m/mod);
}

codeforce 451E

题意：给定以个数的序列，从这个序列中任意取和为s的数，求有多少种取法；
题解：容斥原理，对f[i]进行容斥；（奇加偶减）

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int mod=1e9+7;
ll f[25];
int n;
ll s;
ll inv[30];
ll qm(ll a,ll n)
{
    ll ans=1;
    if(a>=mod)a%=mod;//尤其当a可能爆int时注意先加一个mod
    while(n)
    {
        if(n&1)
        {
            ans=ans*a%mod;
        }
        n>>=1;
        a=a*a%mod;
    }
    return ans;
}
ll C(ll n, ll m)//由于m<=20可以最直接求C(n,m)
{
    if(n<m)return 0;
    ll ans=1;
    for(int i=1;i<=m;i++)
    {
        ans = (n-i+1)%mod*ans%mod*inv[i]%mod;//long long 面前注意写mod，否则可能wa到怀疑人生
    }
    return ans;
}
ll lucas(ll n,ll m)//适合p较小
{
    if(m==0)return 1;
    return C(n%mod,m%mod)*lucas(n/mod,m/mod);
}
ll ans;
void dfs(int cur,ll sum,int flag)
{
    if(sum>s)return ;
    if(cur==(n+1))
    {
        ans+=flag*lucas(s-sum+n-1,n-1);//隔板法
        ans%=mod;
        return ;
    }
    dfs(cur+1,sum,flag);
    dfs(cur+1,sum+f[cur]+1,-flag);
}
int main()
{
    for(int i=1;i<=20;i++)
    {
        inv[i]=qm(i,mod-2);
    }
    scanf("%d%lld",&n,&s);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&f[i]);
    }
    ans=0;
    dfs(1,0,1);
    printf("%d\n",(ans%mod+mod)%mod);
    return 0;
}