计算系数(利用杨辉三角和快速幂)

题目描述

给定一个多项式(by+ax)^k(by+ax)k,请求出多项式展开后x^n \times y^mxn×ym项的系数。

输入输出格式

输入格式:

 

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

 

输出格式:

 

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

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;

ll a,b,k,n,m,arr[1005][1005];

ll pow(ll x,ll y){
    ll ans=1,pas=x;
    while(y){
        if(y&1)
            ans=ans*pas%10007;
        pas=pas*pas%10007;
        y>>=1;
    }
    return ans%10007;
}

int main(){
    cin>>a>>b>>k>>n>>m;
    arr[0][1]=1;
    arr[0][2]=1;
    for(int i=1;i<k;i++){
        for(int j=1;j<=i+2;j++){
            arr[i][j]=(arr[i-1][j-1]+arr[i-1][j])%10007;
        }
    }
    ll ans = arr[k-1][m+1]*pow(b,m)%10007;
    ans = ans*pow(a,n)%10007;
    cout<<ans%10007;
    return 0;
}

 

    原文作者:杨辉三角问题
    原文地址: https://blog.csdn.net/Endeavor_G/article/details/84072974
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞