Bzoj 2142 gift combination mathematics + Number Theory

题目大意：给定n个物品，分给m个人，每个人拿到wi个礼物，问方案数mod P P不一定为质数

首先我们把剩下的礼物也分给一个人 答案明显不变 w[++m]=n-w1-w2-...-wm

然后就会很方便地得到公式：

ans=C(n,w1)*C(n-w1,w2)*C(n-w1-w2,w3)*...*C(n-w1-w2-...-w_(m-1),wm) mod P

       =n!/w1!/w2!/.../wm! mod P

然后p不是质数 我们把P分解 令P=∏pi^ai

我们分别处理，可以得到一次同余方程组ans%pi^ai=lefti，用中国剩余定理合并一下即可。

然后对于每个pi^ai，我们进行以下处理：

将分子和分母化为x*pi^y的形式

然后分母的x部分与pi互质，可以求逆元，分子分母的y部分直接相减即可

然后我们处理阶乘

以19为例，将19!化为x*pi^y的形式，其中pi=3，ai=2 则有

19!%9=(1*2*3*4*5*6*7*8*9*10*11*12*13*14*15*16*17*18*19) %9

           =(1*2*4*5*7*8*10*11*13*14*16*17*19)*3^6*(1*2*3*4*5*6) %9

式子的左半部分是不为3的倍数的数，存在长度为p^a的循环节 求出一个循环节 快速幂处理 然后处理剩余部分

右半部分是6!%9 递归处理即可

我这题解写的真是冷静。。。这题还真TM让人冷静不下来啊-0-

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>using namespace std;typedef long long ll;typedef pair<ll,ll> abcd;ll P,n,m,a[20],ans;struct prime_factor{ll p,a,p_a,left;}prime[50];int tot;void Decomposition(ll x){ll i;for(i=2;i*i<=x;i++)if(x%i==0){prime[++tot].p=i;prime[tot].p_a=1;while(x%i==0)x/=i,prime[tot].a++,prime[tot].p_a*=i;}if(x!=1)prime[++tot].p=x,prime[tot].a=1,prime[tot].p_a=x;}ll Quick_Power(ll x,ll y,ll mo){ll re=1;while(y){if(y&1)re*=x,re%=mo;x*=x,x%=mo;y>>=1;}return re;}abcd Deal(ll x,int pos){ll i;abcd re=make_pair( 1ll , x/prime[pos].p );for(i=1;i<prime[pos].p_a;i++)if(i%prime[pos].p)re.first*=i,re.first%=prime[pos].p_a;re.first=Quick_Power(re.first,x/prime[pos].p_a,prime[pos].p_a);for(i=x-x%prime[pos].p_a+1;i<=x;i++)if(i%prime[pos].p)re.first*=i,re.first%=prime[pos].p_a;if(re.second){abcd temp=Deal(re.second,pos);re.first*=temp.first;re.first%=prime[pos].p_a;re.second+=temp.second;}return re;}abcd EXGCD(ll x,ll y){if(!y)return make_pair(1,0);abcd temp=EXGCD(y,x%y);return make_pair(temp.second,temp.first-x/y*temp.second);}ll Reverse(ll x,ll p){ll re=EXGCD(x,p).first;re=(re%p+p)%p;return re;}void Chinese_Remainder_Theorem(){int i;for(i=1;i<=tot;i++){abcd temp=EXGCD(P/prime[i].p_a,prime[i].p_a);ll x=temp.first;x=(x%P+P)%P;ans+=x*P/prime[i].p_a*prime[i].left;ans%=P;}}int main(){int i,j;ll sum=0;cin>>P>>n>>m;for(i=1;i<=m;i++)cin>>a[i],sum+=a[i];if(sum>n){puts("Impossible");return 0;}if(sum<n)a[++m]=n-sum;Decomposition(P);for(i=1;i<=tot;i++){abcd temp=Deal(n,i);for(j=1;j<=m;j++){abcd _temp=Deal(a[j],i);temp.second-=_temp.second;temp.first*=Reverse(_temp.first,prime[i].p_a);temp.first%=prime[i].p_a;}temp.first*=Quick_Power(prime[i].p,temp.second,prime[i].p_a);prime[i].left=temp.first%prime[i].p_a;}Chinese_Remainder_Theorem();cout<<(ans%P+P)%P<<endl;}

BZOJ 2142 礼物 组合数学+数论