题目大意:给定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; }