题目大意
给定一个多项式 (ax + by)k,请求出多项式展开后 xnym 项的系数
题目解析
运用到杨辉三角和乘方
根据二项式定理,(x+y)k中xmy(k-m)的系数为C(k,m)(杨辉三角第k层,第m项)
转换成 (ax + by)k中xmy(k-m)的系数为C(k,m)*am*b(k-m)
代码
#include<bits/stdc++.h>
#define p 10007
using namespace std;
int a,b,k,m,n;
long long f[1005];
int fun(int x)
{
long long sum=1;
long long l=a%p,r=b%p;
for(int i=1;i<=k-x+1;i++)
sum=(sum*l)%p;
for(int i=1;i<=x-1;i++)
sum=(sum*r)%p;
return (sum*f[x])%p;
}//求a^m与b^(k-m)的乘积再乘C(k,m)
//这里懒得用快速幂,谅解
int main()
{
f[1]=1;
cin>>a>>b>>k>>m>>n;
for(int i=2;i<=k+1;i++)
for(int j=i;j>=1;j--)
f[j]=(f[j-1]+f[j])%p;//求杨辉三角
cout<<fun(n+1);
return 0;
}