欢迎大家访问我的老师的OJ———caioj.cn
题面描述
思路
根据组合计数中的二项式定理,有 ( a x + b y ) k = ∑ i = 0 k C k i a i b k − i x i y k − i \large(ax+by)^k=\sum_{i=0}^kC_k^ia^ib^{k-i}x^iy^{k-i} (ax+by)k=∑i=0kCkiaibk−ixiyk−i
则 x n y m x^ny^m xnym的系数应为 C k n a n b m C^n_ka^nb^m Cknanbm,因为 C k n = k ! n ! ( k − n ) ! \large\ C_k^n=\frac{k!}{n!(k-n)!} Ckn=n!(k−n)!k!,我们可以先求出 k ! k! k!, n ! ( k − n ) ! n!(k-n)! n!(k−n)!,因为这里的模数是质数 10007 10007 10007,
所以我们并不用 exgcd \operatorname{exgcd} exgcd求逆元,而是直接快速幂求出 ( n ! ( k − n ) ! ) 10005 mod 10007 (n!(k-n)!)^{10005}\operatorname{mod }10007 (n!(k−n)!)10005mod10007,即 n ! ( k − n ) ! n!(k-n)! n!(k−n)!的乘法逆元(由费马小定理可以得知)。
剩余的 a n b m a^nb^m anbm,快速幂即可。
AC code
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#define ll long long
using namespace std;
const int mod=10007;
const int N=1e4+10;
ll c[N];int tot;
inline ll pow_mod(ll a,ll b)
{
ll ans=1;a%=mod;
while(b)
{
if(b&1)ans=ans*a%mod;
b>>=1;a=a*a%mod;
}
return ans;
}
inline ll get(ll n)
{
if(tot>=n)return c[n];
while(tot<=n)
{
++tot;
c[tot]=c[tot-1]*tot%mod;
}
return c[n];
}
int main()
{
ll a,b,k,n,m;scanf("%lld%lld%lld%lld%lld",&a,&b,&k,&n,&m);
tot=0;c[0]=1;
ll x=get(k),y=get(n)*get(m);//y=get(n)+get(k-n);
y=pow_mod(y,mod-2);
ll s=x*y%mod;
x=pow_mod(a,n),y=pow_mod(b,m);
s=s*x*y%mod;
printf("%lld\n",s);
return 0;
}