[NOIP2011 提高组] 计算系数

[NOIP2011 提高组] 计算系数

题目描述

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

输入格式

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

输出格式

输出共一行，包含一个整数，表示所求的系数。

这个系数可能很大，输出对 $10007$ 取模后的结果。

样例 #1

样例输入 #1

1 1 3 1 2

样例输出 #1

3

提示

【数据范围】

对于 $30\%$ 的数据，有 $ 0\le k\le 10$。

对于 $50\%$ 的数据，有 $ a=1$，$b=1$。

对于 $100\%$ 的数据，有 $0\le k\le 1000$，$0\le n,m\le k$，$n+m=k$，$0\le a,b\le 10^6$。

noip2011 提高组 day2 第 1 题。

有关计算二项式系数的模板题（杨辉三角）

AC代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=10007;
const int M=10086;
int fac[M];
int inv[M];
int read(){
	int ans=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		f*=(ch=='-')?-1:1;
		ch=getchar();
	}
	while(isdigit(ch)){
		ans=ans*10+ch-'0';
		ch=getchar();
	}
	return ans*f;
}

int fastPow(int a,int n){
    int ans1=1;
    a%=mod;
    while(n){
        if(n&1)ans1=ans1*a%mod;  
        a=a*a%mod;        
        n>>=1;   
    }
    return ans1;
}

int C(int n,int m){
	return (fac[n]*inv[m]%mod*inv[n-m]%mod)%mod;
}


signed main(){
	int a=read(),b=read(),k=read(),n=read(),m=read(),ans_;
	fac[0]=1;
	for(int i=1;i<=n+m;i++){
		fac[i]=(fac[i-1]*i)%mod;
		inv[i]=fastPow(fac[i],mod-2);
	}
	ans_=(fastPow(a,n)%mod*fastPow(b,m)%mod*C(k,n)%mod)%mod;
		printf("%lld",ans_);
	}