洛谷 P1313 计算系数

题目描述

给定一个多项式(by+ax)^k，请求出多项式展开后x^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题

---

杨辉三角+快速幂
一开始想用组合数公式直接算，但好像会炸。。。
改递推，但又忘了用long long。。。

---

#include<iostream>
#include<cstdio>
using namespace std;
const long long md=10007;
int a,b,k,n,m;
long long ans,f[1005][1005];
long long pow(int x,int p)//快速幂 
{
    if(p==0)
        return 1;
    long long tmp=pow(x,p/2)%md;
    tmp=(tmp*tmp)%md;
    if(p%2==1)
        tmp=(tmp*x)%md;
    return tmp;
}
int main()
{
    scanf("%d%d%d%d%d",&a,&b,&k,&n,&m);
    ans=(pow(a,n)*pow(b,m))%md;
    for(int i=1;i<=k+1;i++)
        for(int j=1;j<=i;j++)
        {
            if(j==1)
                f[i][j]=1;
            else
                f[i][j]=(f[i-1][j-1]+f[i-1][j])%md;
        }
    ans*=f[k+1][m+1];
    ans%=md;
    printf("%lld\n",ans);
    return 0;
}