1. 程式人生 > >【基礎練習】【組合數+快速冪】codevs1137 計算係數題解

【基礎練習】【組合數+快速冪】codevs1137 計算係數題解

非常簡單的一道題目 可是編譯器總是出問題= =

轉載請註明出處 CSDN ametake版權所有

題目

題目描述 Description

給定一個多項式(ax + by)^k,請求出多項式展開後x^n y^m項的係數。

輸入描述 Input Description

共一行,包含 5 個整數,分別為a,b,k,n,m,每兩個整數之間用一個空格隔開。

輸出描述 Output Description

輸出共 1 行,包含一個整數,表示所求的係數,這個係數可能很大,輸出對10007 取模後的結果。

樣例輸入 Sample Input

1 1 3 1 2

樣例輸出 Sample Output

3

資料範圍及提示 Data Size & Hint 資料範圍
對於 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。

直接上程式碼

//codevs1137 計算係數 快速冪 NOIP2011TG
//copyright by ametake
#include
#include
using namespace std;

const long long maxn=10007;
int aa[1000+10][1000+10];

int mi(long long a,long long b,long long c)
{
	long long ans=1;
	while (b)
	{
		if (b & 1==1) ans=(ans*a)%c;
		a=(a*a)%c;
		b>>=1;
	}
	return ans;
}

int yanghui(long long x,long long y)
{
	for (int i=0;i<=x;i++){
		aa[i][0]=1;aa[i][i]=1;
	}
	for (int i=1;i<=x;i++){
		for (int j=1;j<=i-1;j++){
			aa[i][j]=(aa[i-1][j]+aa[i-1][j-1])%maxn;
		}
	}
	return aa[x][y];
}

int main()
{
	long long a = 0,b=0,k=0,n=0,m=0;
	scanf("%lld%lld%lld%lld%lld",&a,&b,&k,&n,&m);
	long long fin=((mi(a,n,maxn)%maxn)*(mi(b,m,maxn)%maxn)*(yanghui(k,n)%maxn))%maxn;
	printf("%lld",fin);
}

——休對故人思故國,且將新火試新茶。詩酒趁年華。