计算系数 (vijos P1739)

给定一个多项式(ax + by)^k，请求出多项式展开后x^n * y^m项的系数。

格式

输入格式

共一行，包含5个整数，分别为a，b，k，n，m，每两个整数之间用一个空格隔开。输出格式

输出共1行，包含一个整数，表示所求的系数，这个系数可能很大，**输出对10007取模后的结果**。

样例1

样例输入1

1 1 3 1 2

样例输出1

3

限制

1s

 

提示

对于30%的数据，有0 ≤ k ≤ 10；

 对于50%的数据，有a = 1, b = 1；

 对于100%的数据，有0 ≤ k ≤ 1000，0 ≤ n, m ≤ k，且n+m = k，0 ≤ a，b ≤ 1,000,000.

分析：

用杨辉三角形刚好可以计算系数。

#include<cstdio>
#include<cstring>
using namespace std;
int xs1[1100],xs2[1100];
int xs[1100];
void getXs(int k)//计算杨辉三角形的第k行
{
	int i,s;
	xs[0]=xs[2]=1;
	xs[1]=2;
	if(k==2)
		return;
	for(i=3;i<=k;i++)
	{
		xs1[0]=xs1[i]=1;
		for(s=1;s<i;s++)
			xs1[s]=(xs[s-1]+xs[s])%10007;
		memcpy(xs,xs1,1100*sizeof(int));
	}
}

int getPower(int s, int t)//s的t次幂
{
	int val; 
	int i;
	if(t==0)
		return 1;
	s=s%10007;   //如果不做此步，先乘后取模，会溢出
	val=s;
	for(i=0;i<t-1;i++)
		val=(val*s)%10007;
	return val;
}

int main()
{
	int val,x1,x2;
	int a,b,k,n,m;
	scanf("%d%d%d%d%d",&a,&b,&k,&n,&m);
	if(k==0)
	{
		printf("1");
	}
	else if(k==1)
	{
		if(m==0)
			printf("%d",a);
		else
			printf("%d",b);
	}
	else
	{
		getXs(k);
		val=xs[m];
		x1=getPower(a,n);
		x2=getPower(b,m);
		val=(val*x1)%10007;
		val=(val*x2)%10007;
		printf("%d",val);
	}
	return 0;
}