Description
给定一个多项式 (ax + by)^k ,请求出多项式展开后 xnym 项的系数。
Input
共一行,包含 5 个整数,分别为 a,b,k,n,m,每两个整数之间用一个空格隔开。
Output
输出共 1 行,包含一个整数,表示所求的系数,这个系数可能很大,输出对 10007 取模后的结果。
Sample Input
1 1 3 1 2
Sample Output
3
Data Constraint
对于 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<cmath>
#include<cstdio>
#include<iostream>
using namespace std;
int a,b,k,m,n,ans=1;
int f[1005][1005];
int main(){
cin>>a>>b>>k>>n>>m;
for (int i=1;i<=k;i++){
f[i][1]=1;
f[i][i+1]=1;
}
for (int i=2;i<=k;i++)
for (int j=2;j<=i;j++)
f[i][j]=(f[i-1][j]+f[i-1][j-1])%10007;
a=a%10007; b=b%10007;
for (int i=1;i<=n;i++)
ans=(ans*a)%10007;
for (int i=1;i<=m;i++)
ans=(ans*b)%10007;
ans=(ans*f[k][k-n+1])%10007;
cout<<ans<<endl;
return 0;
}