P2183 [国家集训队]礼物 （扩展卢卡斯）

题目连接

题面：

题解：

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<vector>
#define ll long long
using namespace std;
const int maxn=100100;
int prime[maxn];
bool ha[maxn]={false};
int cnt=0;
ll mod;

ll mypow(ll a,ll b,ll p)
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=(ans*a)%p;
        a=(a*a)%p;
        b>>=1;
    }
    return ans;
}

void Prime(void)
{
    ha[1]=true;
    for(int i=2;i<maxn;i++)
    {
        if(!ha[i]) prime[cnt++]=i;
        for(int j=0;j<cnt&&prime[j]*i<maxn;j++)
        {
            ha[i*prime[j]]=true;
            if(i%prime[j]==0) break;
        }
    }
    return ;
}

ll pp[20],e[20],sum[20];
int res=0;

void only(ll n)
{

    for(int i=0;i<cnt&&prime[i]*prime[i]<=n;i++)
    {

        if(n%prime[i]) continue;

        pp[++res]=prime[i];
        sum[res]=1;
        while(n%prime[i]==0)
        {
            sum[res]*=prime[i];
            e[res]++;
            n/=prime[i];
        }
        if(n==1) break;
    }
    if(n>1) pp[++res]=n,e[res]=1,sum[res]=n;
    return ;
}

ll cal(ll n,ll p)
{
    ll ans=0;
    while(n)
    {
        ans+=n/p;
        n/=p;
    }
    return ans;
}

ll fac(ll n,ll p,ll pk)
{
    if(n==0) return 1;
    ll ans=1;
    for(int i=1;i<=pk;i++)
    {
        if(i%p) ans=(ans*i)%pk;
    }
    ans=mypow(ans,n/pk,pk);
    for(int i=1;i<=n%pk;i++)
    {
        if(i%p)
            ans=ans*i%pk;
    }
    return ans*fac(n/p,p,pk)%pk;
}

ll exgcd(ll a,ll b,ll &x,ll &y)
{
    if(b==0)
    {
        x=1,y=0;
        return a;
    }
    ll d=exgcd(b,a%b,x,y);
    ll z=x;
    x=y;
    y=z-a/b*y;
    return d;
}

ll inv(ll n,ll p)
{
    ll x,y;
    ll d=exgcd(n,p,x,y);
    return (x%p+p)%p;
}

ll C(ll n,ll m,ll p,ll pk)
{
    if(n<m) return 0;
    ll nn=fac(n,p,pk),mm=fac(m,p,pk),nm=fac(n-m,p,pk);
    ll cnt=cal(n,p)-cal(m,p)-cal(n-m,p);
    return nn*inv(mm,pk)%pk*inv(nm,pk)%pk*mypow(p,cnt,pk)%pk;

}

ll Lchina(ll n,ll m, ll p)
{
    ll ans=0;
    for(int i=1;i<=res;i++)
    {
        ans=(ans+C(n,m,pp[i],sum[i])*p/sum[i]%p*inv(p/sum[i],sum[i])%p)%p;
    }
    return ans;
}

int main(void)
{
    ll n,m;
    ll p;
    Prime();
    scanf("%lld%lld%lld",&p,&n,&m);
    only(p);

    ll num[10]={0},a[10]={0};
    for(int i=1;i<=m;i++)
    {
        scanf("%lld",&a[i]);
        num[i]=num[i-1]+a[i];
    }
    ll ans=1;
    if(n<num[m])
    {
        printf("Impossible\n");
        return 0;
    }
    for(int i=0;i<m;i++)
    {
        ans=(ans*Lchina(n-num[i],a[i+1],p))%p;
    }
    printf("%lld\n",ans);
    return 0;
}