【洛谷】P1313 计算系数（快速幂+杨辉三角）

题目

题目描述

给定一个多项式(by+ax)n*y^m 项的系数。

输入输出格式

输入格式：

输入文件名为factor.in。

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

输出格式：

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

输入输出样例

输入样例#1：

1 1 3 1 2

输出样例#1：

3

说明

数据范围

对于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。

noip2011提高组day2第1题

 

 

---

 

 

分析

我太蒟了，先了一发，只拿了20，发现结果是先乘再取的模，中间可能会爆，改了下，交了上去，50。看了看，快速幂基数没取模，再改，80。 还没有A？int全部改long long，终于A了。

 

 

---

 

 

代码

 

#include <cstdio>

#include <algorithm>

using namespace std;

typedef long long ll;

ll MOD=10007,C[1007][1007];

ll poww(ll a,ll b)

{

	ll base=a,ans=1;

	while(b!=0)

	{

		if(b&1) ans=(ans*base)%MOD;

		base=(base*base)%MOD;

		b=b>>1;

	}

	return ans;

}

int main()

{

	ll n,m,a,b,k,x,y;

	scanf("%lld%lld%lld%lld%lld",&a,&b,&k,&n,&m);

	C[0][0]=1;C[1][1]=1;

	for(ll i=2;i<=k+1;i++)

	for(ll j=1;j<=i;j++)

	{

		C[i][j]=C[i-1][j-1]+C[i-1][j];

		C[i][j]%=MOD;

	}

	ll ans=(poww(a,n)*poww(b,m))%MOD*C[k+1][k+1-n];

	ans%=MOD;

	printf("%lld",ans);

	return 0;

}