题目
给定一个多项式$ (ax + by)^k$ ,请求出多项式展开后 $xnym $项的系数。
分析
根据二项式定理,有 ( a x + b y ) k = ∑ i = 0 k C k i a i b k − i x i y k − i (ax+by)^k=\sum_{i=0}^kC_k^ia^ib^{k-i}x^iy^{k-i} (ax+by)k=i=0∑kCkiaibk−ixiyk−i,所以 x n y m x^ny^m xnym项的系数为 C k a n b m C_ka^nb^m Ckanbm,然后通过快速幂和乘法逆元求出答案。
代码
#include <cstdio>
#define mod 10007
using namespace std;
typedef long long ll;
ll a,b,k,n,m,ans;
ll ksm(ll x,ll y){
ll ans=1;
while (y){
if (y&1) ans=ans*x%mod;
x=x*x%mod; y>>=1;
}
return ans;
}
ll c(ll n,ll m){
ll ans=1;
for (ll i=n-m+1;i<=n;i++) ans=ans*i%mod;
for (ll i=1;i<=m;i++) ans=ans*ksm(i,mod-2)%mod;
return ans;
}
int main(){
scanf("%lld%lld%lld%lld%lld",&a,&b,&k,&n,&m);
ans=ksm(a,n)*ksm(b,m)%mod*c(k,n)%mod;
return !printf("%lld",ans);
}