题目描述 Description
给定一个多项式(ax + by)^k,请求出多项式展开后x^n y^m项的系数。
输入描述 Input Description
共一行,包含 5 个整数,分别为a,b,k,n,m,每两个整数之间用一个空格隔开。
输出描述 Output Description
输出共 1 行,包含一个整数,表示所求的系数,这个系数可能很大,输出对10007 取模后的结果。
数据范围
对于 30%的数据,有0≤k≤10;
对于 50%的数据,有a = 1,b = 1;
对于 100%的数据,有0≤k≤1,000,0≤n, m≤k,且n + m = k,0≤a,b≤1,000,000。
扔个二项式定理的板子= =。
#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
const ll ha=10007;
ll a,b,k,n,m;
ll ksm(ll x,ll y)
{
if(y==0) return 1;
if(y==1) return x%ha;
ll hah=ksm(x%ha,y>>1);
if(y&1) return ((((hah%ha)*(hah%ha))%ha)*x)%ha;
else return ((hah%ha)*(hah%ha))%ha;
}
ll numa[1500];
ll numb[1500];
void fj(ll x)
{
ll tot=2;
while(x!=1)
{
if(x%tot==0)
{
numa[tot]++;
x/=tot;
}
else
tot++;
}
}
void fj1(ll x)
{
ll tot=2;
while(x!=1)
{
if(x%tot==0)
{
numb[tot]++;
x/=tot;
}
else
tot++;
}
}
void C(ll n,ll m)
{
for(ll i=n+1;i<=m;i++)
fj(i);
for(ll i=1;i<=m-n;i++)
fj1(i);
}
int main()
{
scanf("%lld%lld%lld%lld%lld",&a,&b,&k,&n,&m);
ll hah=(ksm(a%ha,n)*ksm(b%ha,m))%ha;
C(n,k);
ll ans1=1;
for(ll i=1;i<=k;i++)
{
numa[i]-=numb[i];
ans1%=ha;
ans1*=ksm(i,numa[i]);
}
ans1%=ha;
printf("%lld",(ans1*hah)%ha);
return 0;
}