bzoj 3129: [Sdoi2013]方程

21 篇文章 0 订阅
5 篇文章 0 订阅

题意:

给你一个方程: x1+x2+x3+xn=m 的整数解个数。
其中前 n1 个数要 <=ai <script type="math/tex" id="MathJax-Element-123"><=a_i</script>后 n2 个数 >=ai+n1

题解:

要大于某个数,就直接m减去 ai1 就好了。
然后用隔板法,在容斥一下,就可以了。
至于为什么可以扩展lucas,不用管的了反正就可以了
code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
LL MOD;
LL p_i[25],p_k[25],num=0;
void pre()
{
    LL x=MOD;
    for(LL i=2;i*i<=x;i++)
        if(x%i==0)
        {
            LL k=1;
            while(x%i==0) x/=i,k*=i;
            p_i[++num]=i;p_k[num]=k;
        }
    if(x!=1) p_i[++num]=x,p_k[num]=x;
}
LL pow(LL a,LL b,LL mod)
{
    LL ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;b>>=1;
    }
    return ans;
}
LL inv(LL a,LL pi,LL pk)
{
    if(!a) return 1;
    return pow(a,pk-(pk/pi)-1,pk);
}
LL mul(LL n,LL pi,LL pk)
{
    if(!n) return 1;
    LL ans=1;
    for(LL i=2;i<=pk;i++)
        if(i%pi) ans=ans*i%pk;
    ans=pow(ans,n/pk,pk);
    for(LL i=2;i<=n%pk;i++)
        if(i%pi) ans=ans*i%pk;
    return ans*mul(n/pi,pi,pk)%pk;
}
LL C(LL n,LL m,LL mod,LL pi,LL pk)
{
    if(m>n) return 0;
    LL a=mul(n,pi,pk),b=mul(m,pi,pk),c=mul(n-m,pi,pk);
    LL ans,k=0;
    for(LL i=n;i;i/=pi) k+=i/pi;
    for(LL i=m;i;i/=pi) k-=i/pi;
    for(LL i=n-m;i;i/=pi) k-=i/pi;
    ans=a*inv(b,pi,pk)%pk*inv(c,pi,pk)%pk*pow(pi,k,pk)%pk;
    return ans*(mod/pk)%mod*inv(mod/pk,pi,pk)%mod;
}
LL lucas(LL n,LL m)
{
    if(m>n) return 0;
    LL ans=0;
    for(LL i=1;i<=num;i++) ans=(ans+C(n,m,MOD,p_i[i],p_k[i]))%MOD;
    return ans;
}
LL n,m,n1,n2,a[15];
LL ans;
void dfs(LL x,LL k,LL num)
{
    if(k<n) return;
    if(x>n1)
    {
        if(num&1) ans-=lucas(k-1,n-1);
        else ans+=lucas(k-1,n-1);
        ans%=MOD;
        return;
    }
    dfs(x+1,k,num);
    dfs(x+1,k-a[x],num+1);
}
int main()
{
    LL T;scanf("%lld %lld",&T,&MOD);
    pre();
    while(T--)
    {
        scanf("%lld %lld %lld %lld",&n,&n1,&n2,&m);
        for(LL i=1;i<=n1;i++) scanf("%lld",&a[i]);
        for(LL i=1;i<=n2;i++)
        {
            LL x;scanf("%lld",&x);
            m-=(x-1);
        }
        ans=0;
        dfs(1,m,0);
        printf("%lld\n",(ans+MOD)%MOD);
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值