zoukankan      html  css  js  c++  java
  • BZOJ2142礼物——扩展卢卡斯

    题目描述

    一年一度的圣诞节快要来到了。每年的圣诞节小E都会收到许多礼物,当然他也会送出许多礼物。不同的人物在小E
    心目中的重要性不同,在小E心中分量越重的人,收到的礼物会越多。小E从商店中购买了n件礼物,打算送给m个人
    ,其中送给第i个人礼物数量为wi。请你帮忙计算出送礼物的方案数(两个方案被认为是不同的,当且仅当存在某
    个人在这两种方案中收到的礼物不同)。由于方案数可能会很大,你只需要输出模P后的结果。

    输入

    输入的第一行包含一个正整数P,表示模;
    第二行包含两个整整数n和m,分别表示小E从商店购买的礼物数和接受礼物的人数;
    以下m行每行仅包含一个正整数wi,表示小E要送给第i个人的礼物数量。

    输出

    若不存在可行方案,则输出“Impossible”,否则输出一个整数,表示模P后的方案数。

    样例输入

    100
    4 2
    1
    2

    样例输出

    12
    【样例说明】
    下面是对样例1的说明。
    以“/”分割,“/”前后分别表示送给第一个人和第二个人的礼物编号。12种方案详情如下:
    1/23 1/24 1/34
    2/13 2/14 2/34
    3/12 3/14 3/24
    4/12 4/13 4/23
    【数据规模和约定】
    设P=p1^c1 * p2^c2 * p3^c3 * … *pt ^ ct,pi为质数。
    对于100%的数据,1≤n≤109,1≤m≤5,1≤pi^ci≤10^5。
     
    如果$sum w_{i}>n$显然无解,如果$sum w_{i}<n$可以将最后剩下那些礼物看作给第$m+1$个人。那么答案就是$C_{n}^{w_{1}}*C_{n-w_{1}}^{w_{2}}*...*C_{w_{m+1}}^{w_{m+1}}$,展开之后就是$frac{n!}{(w_{1})!(w_{2})!...(w_{m+1})!}$,因为模数不一定是质数,所以用扩展卢卡斯求一下即可。
    #include<set>
    #include<map>
    #include<queue>
    #include<stack>
    #include<cmath>
    #include<cstdio>
    #include<vector>
    #include<bitset>
    #include<cstring>
    #include<iostream>
    #include<algorithm>
    #define ll long long
    using namespace std;
    ll n,P;
    int m;
    ll w[10];
    ll quick(ll x,ll y,ll mod)
    {
        ll res=1ll;
        while(y)
        {
            if(y&1)
            {
                res=res*x%mod;
            }
            x=x*x%mod;
            y>>=1;
        }
        return res;
    }
    ll find(ll n,ll p,ll mod)
    {
        if(!n)
        {
            return 1ll;
        }
        ll res=1ll;
        for(ll i=2;i<=mod;i++)
        {
            if(i%p)
            {
                res*=i,res%=mod;
            }
        }
        res=quick(res,n/mod,mod);
        for(ll i=2;i<=n%mod;i++)
        {
            if(i%p)
            {
                res*=i,res%=mod;
            }
        }
        return res*find(n/p,p,mod)%mod;
    }
    ll inv(ll n,ll mod,ll p)
    {
        ll phi=mod-(mod/p);
        return quick(n,phi-1,mod);
    }
    ll CRT(ll b,ll mod,ll p)
    {
        return b*inv(P/mod,mod,p)%P*(P/mod)%P;
    }
    ll C(ll n,int m,ll p,ll mod)
    {
        ll res=find(n,p,mod);
        ll k=0;
        for(ll i=n;i;i/=p)
        {
            k+=i/p;
        }
        for(int i=1;i<=m;i++)
        {
            res*=inv(find(w[i],p,mod),mod,p),res%=mod;
            for(ll j=w[i];j;j/=p)
            {
                k-=j/p;
            }
        }
        return res*quick(p,k,mod)%mod;
    }
    ll ex_lucas(ll n,int m)
    {
        ll res=0;
        ll sum=P;
        ll mod;
        for(int i=2;1ll*i*i<=P;i++)
        {
            if(sum%i==0)
            {
                mod=1ll;
                while(sum%i==0)
                {
                    mod*=i;
                    sum/=i;
                }
                res+=CRT(C(n,m,i,mod),mod,i),res%=P;
            }
        }
        if(sum!=1)
        {
            res+=CRT(C(n,m,sum,sum),sum,sum),res%=P;
        }
        return res;
    }
    int main()
    {
        scanf("%lld",&P);
        scanf("%lld%d",&n,&m);
        ll sum=n;
        for(int i=1;i<=m;i++)
        {
            scanf("%lld",&w[i]);
            sum-=w[i];
        }
        if(sum<0)
        {
            printf("Impossible");
            return 0;
        }
        w[++m]=sum;
        printf("%lld",ex_lucas(n,m));
    }
  • 相关阅读:
    【Codevs 2630】宝库通道
    【Codevs 2115】数集分割
    【HDU2037】今年暑假不AC
    【Codeforces】Round #376 (Div. 2)
    【Dairy】2016.10.17-1 OIer最悲剧的事情
    【Codevs 3115】高精度练习之减法
    【Codevs1080】质数环
    【T^T 1871】获取敌情
    【Codevs3151】交通管制I
    【Codeforces】716D Complete The Graph
  • 原文地址:https://www.cnblogs.com/Khada-Jhin/p/10382361.html
Copyright © 2011-2022 走看看